【中等】
给你一个链表的头节点 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]
提示:
- 链表中节点的数目在范围
[0, 200]
内 -100 <= Node.val <= 100
-200 <= x <= 200
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode() {}
* ListNode(int val) { this.val = val; }
* ListNode(int val, ListNode next) { this.val = val; this.next = next; }
* }
*/
class Solution {
public ListNode partition(ListNode head, int x) {
if (head == null) {
return null;
}
// 左边链表保存小于x的节点
ListNode leftDummy = new ListNode(-1);
// 右边链表保存大于等于x的节点
ListNode rightDummy = new ListNode(-1);
ListNode left = leftDummy;
ListNode right = rightDummy;
ListNode p = head;
// 遍历原链表,拆分成两个链表
while (p != null) {
if (p.val < x) {
left.next = p;
left = left.next;
} else {
right.next = p;
right = right.next;
}
p = p.next;
}
// 因为右侧链表节点仍然可能连着原列表的节点,因此需要将next置为null
right.next = null;
// 连接两个链表
left.next = rightDummy.next;
return leftDummy.next;
}
}