LeetCode 86. 分隔链表
题目描述
✅ 86. 分隔链表
思路分析
我们可以使用两个虚拟头节点来分别存储小于 x 和大于等于 x 的节点。
遍历链表时,将节点根据值的大小分别添加到这两个链表中。
最后,将两个链表连接起来,返回小于 x 的链表的头节点。
参考代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
func partition(head *ListNode, x int) *ListNode {
if head == nil || head.Next == nil {
return head
}
leftHead := &ListNode{}
rightHead := &ListNode{}
left := leftHead
right := rightHead
cur := head
for cur != nil {
if cur.Val < x {
left.Next = cur
left = left.Next
} else {
right.Next = cur
right = right.Next
}
cur = cur.Next
}
right.Next = nil // 特别注意:此处一定要断开,防止循环引用。
left.Next = rightHead.Next
return leftHead.Next
}
1
write your code here
CC BY-NC-SA 4.0
许可协议,转载请注明出处!
本博客所有文章除特别声明外,均采用