86.分隔链表

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

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

示例 1:

img

1
2
输入:head = [1,4,3,2,5,2], x = 3
输出:[1,2,2,4,3,5]

示例 2:

1
2
输入:head = [2,1], x = 2
输出:[1,2]

Solution

两个链表模拟就行了

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
29
30
31
32
class Solution {
public ListNode partition(ListNode head, int x) {
// 创建两个虚拟头节点,用于构建小于x和大于等于x的两个链表
ListNode small = new ListNode(0);
ListNode smallHead = small; // smallHead用于保留小于x链表的头部引用
ListNode large = new ListNode(0);
ListNode largeHead = large; // largeHead用于保留大于等于x链表的头部引用

// 遍历原始链表,根据节点值与x的比较结果,分别添加到两个链表中
while (head != null) {
if (head.val < x) {
// 如果当前节点值小于x,添加到small链表
small.next = head;
small = small.next;
} else {
// 如果当前节点值大于等于x,添加到large链表
large.next = head;
large = large.next;
}
head = head.next; // 移动到下一个节点
}

// 将大于等于x的链表的末尾设为null,防止形成环
large.next = null;
// 将小于x的链表和大于等于x的链表连接起来
small.next = largeHead.next;

// 返回重新组合后的链表的头节点
return smallHead.next;
}
}