86. 分隔链表-双指针解法
题目描述:
给你一个链表的头节点 head 和一个特定值 x ,请你对链表进行分隔,使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
你应当 保留 两个分区中每个节点的初始相对位置。
题解(双指针):
// 解法1
public ListNode partition(ListNode head, int x) {
// 涉及两个新链表-虚拟头节点
ListNode dummy1 = new ListNode(-1);
ListNode dummy2 = new ListNode(-1);
ListNode p1 = dummy1;
ListNode p2 = dummy2;
ListNode p = head;
while (p != null) {
if (p.val < x) {
p1.next = p;
p1 = p1.next;
} else {
p2.next = p;
p2 = p2.next;
}
p = p.next;
}
// 最后,设置尾指针的 next 为空
p1.next = null;
p2.next = null;
// 连接两个链表
p1.next = dummy2.next;
return dummy1.next;
}
// 解法2
public ListNode partition2(ListNode head, int x) {
// 涉及两个新链表-虚拟头节点
ListNode dummy1 = new ListNode();
ListNode dummy2 = new ListNode();
ListNode p1 = dummy1;
ListNode p2 = dummy2;
ListNode p = head, r;
// 两个新链表 瓜分 head
while (p != null) {
// 记录当前节点的下一个,防止断链
r = p.next;
if (p.val < x) {
p1.next = p;
p1 = p1.next;
} else {
p2.next = p;
p2 = p2.next;
}
// 已经被分配的节点,断开原链表
p.next = null;
p = r;
}
// 连接两个链表
p1.next = dummy2.next;
return dummy1.next;
}