[LeetCode]86. 分隔链表(Partition List)

题目描述

给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。

你应当保留两个分区中每个节点的初始相对位置。

示例:

输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5

解题思路

双指针。

代码

Python 3.6

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# 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:
left = ListNode(0)
right = ListNode(100)
p1, p2 = left, right
cur = head
while cur is not None:
if cur.val < x:
p1.next = cur
p1 = cur
else:
p2.next = cur
p2 = cur
cur = cur.next
p2.next = None
p1.next = right.next
return left.next

执行用时 : 76 ms
内存消耗 : 13.8 MB