给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
# Definition for singly-linked list. # class ListNode: # def __init__(self, x): # self.val = x # self.next = None class Solution: def partition(self, head: ListNode, x: int) -> ListNode: firstNode=ListNode(0) firstNode.next=head sepNode=firstNode preNode=firstNode curNode=head while curNode is not None: if curNode.val<x: if preNode is sepNode: preNode=curNode sepNode=curNode curNode=curNode.next else: preNode.next=curNode.next curNode.next=sepNode.next sepNode.next=curNode sepNode=curNode curNode=preNode.next else: preNode=curNode curNode=preNode.next return firstNode.next