题目:
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
思路:
把小于x的节点从原链表删除,然后依次串起来,就得到2个链表,一个是小于x的链表,另一个是大于等于x的链表,然后把这俩链表串起来即可。
代码:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def partition(self, head, x):
"""
:type head: ListNode
:type x: int
:rtype: ListNode
"""
aim = ListNode(-1)
aim.next = head
pre = aim
first = None
last = None
while head:
if head.val < x:
temp = head
head = head.next
pre.next = head
temp.next = None
if not first:
first = temp
last = temp
else:
last.next = temp
last = last.next
else:
head = head.next
pre = pre.next
if last:
last.next = aim.next
return first if first else aim.next