LeetCode 86. 分隔链表

题目描述

86. 分隔链表

image-20230313213650579

思路分析

我们可以使用两个虚拟头节点来分别存储小于 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
}

➡️ 点击查看 Java 题解

1
write your code here
本文作者:
本文链接: https://hgnulb.github.io/blog/2023/63245444
版权声明: 本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议,转载请注明出处!