2022.7.1 分隔链表

---

一、86. 分隔链表

1.题目

跳转LeetCode

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

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

2.分析

参考文章

题目给的函数签名为：

public ListNode partition(ListNode head, int x) {

}

把原链表分成两个小链表，一个链表中的元素大小都小于 x，另一个链表中的元素都大于等于 x，最后再把这两条链表接到一起，就得到了题目想要的结果。

具体编程的思路为：
①为两个小链表设置虚拟的头结点；
②设置用于遍历、拓展小链表的指针，指针初始指向虚拟头结点；
③设置用于遍历原链表的指针，放入 while 循环中遍历；
④对于原链表中的每个节点，根据其与所给 x 的大小关系，在相应的小链表中添加与正遍历到的节点数值相同的新节点（与题解做法不同，题解中的做法可以抽象理解为将原链表中的节点移动到小链表中）；
⑤当完成一次④中操作后，小链表&原链表 对应的指针都要向后移一位；
⑥重复④⑤直至原链表全部遍历，将元素都大于等于 x的小链表接到元素都小于 x的链表后面即可。

3.代码

package LinkedList;

/**
 * @author: LYZ
 * @date: 2022/7/1 16:28
 * @description: 给你一个链表的头节点 head 和一个特定值 x ，请你对链表进行分隔，
 * 使得所有 小于 x 的节点都出现在 大于或等于 x 的节点之前。
 * 你应当 保留 两个分区中每个节点的初始相对位置。
 */
public class Partition {
    public static void main(String[] args) {
        ListNode head = new ListNode(1, new ListNode(4, new ListNode(3, new ListNode(2, new ListNode(5, new ListNode(2))))));
        Partition par = new Partition();
        ListNode ans = par.partition(head, 3);

        while (ans != null) {
            System.out.println(ans.val);
            ans = ans.next;
        }
    }

    public ListNode partition(ListNode head, int x) {
        // 存放小于 x 的链表的虚拟头结点
        ListNode dummySmall = new ListNode(-1);
        // 存放大于等于 x 的链表的虚拟头结点
        ListNode dummyBig = new ListNode(-1);
        // 负责生成结果链表的指针
        ListNode pointSmall = dummySmall;
        ListNode pointBig = dummyBig;
        // 负责遍历原链表
        ListNode pointHead = head;

        while (pointHead != null) {
            if (pointHead.val < x) {
                pointSmall.next = new ListNode(pointHead.val);
                pointSmall = pointSmall.next;
            } else {
                pointBig.next = new ListNode(pointHead.val);
                pointBig = pointBig.next;
            }
            pointHead = pointHead.next;
        }

        pointSmall.next = dummyBig.next;

        return dummySmall.next;
    }
}

总结

可以与 21. 合并两个有序链表 一起分析，体会虚拟头结点存在的意义。

当做题时需创建新链表时，为新链表使用虚拟头结点，并设置遍历该链表的指针。