题目描述:
给你一个链表的头节点 head
和一个特定值 x
,请你对链表进行分隔,使得所有 小于 x
的节点都出现在 大于或等于 x
的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
示例 1:
输入:head = [1,4,3,2,5,2], x = 3 输出:[1,2,2,4,3,5]
示例 2:
输入:head = [2,1], x = 2 输出:[1,2]
解题思路:这道题的主要是用一个x 值分割了这个链表,要将小于x 的节点放在前面,大于等于x 的放后面,但相对位置不能变,也就是只用在遍历时进行分割,而无需进行排序
所以需要两个空链表,将小于x 的放在第一个链表,大于等于x 的放在第二个链表,然后将第一和第二个链表合并即可
代码:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution(object):
def partition(self, head, x):
"""
:type head: ListNode
:type x: int
:rtype: ListNode
"""
dummy1 = ListNode(-1)
dummy2 = ListNode(-1)
p = head
p1 = dummy1
p2 = dummy2
while (p):
if (p.val < x):
p1.next = p
p1 = p1.next
else:
p2.next = p
p2 = p2.next
temp = p.next
p.next = None
p = temp
p1.next = dummy2.next
return dummy1.next
有一个需要注意的地方:p指针不能每次直接用p.next 来向后挪,如果这样会形成一个环