分隔链表-双指针

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;
}