分隔链表

目录

• 题目描述
• 解题思路
• 解题代码

题目描述

1. 题目地址：https://leetcode.cn/problems/partition-list/
2. 题目要求
   给你一个链表的头节点head和一个特定值 x ，请你对链表进行分隔，使得所有 小于 x 的节点都出现在 **大于或等于 **x 的节点之前。
   你应当 保留 两个分区中每个节点的初始相对位置。
3. 示例 1：
   输入：head = [1,4,3,2,5,2], x = 3
   输出：[1,2,2,4,3,5]
   示例2:
   输入：head = [2,1], x = 2
   输出：[1,2]

解题思路

1. 只需维护两个链表small 和large 即可,将 small 链表尾节点指向large 链表的头节点即可完成对链表的分隔
2. 同时设small和large节点指向当前链表的末尾节点，开始时smallHead=small,largeHead=large
3. 定义 smallHead 和 largeHead 为两个链表的哑节点，它们的 next 指针指向链表的头节点，目的是为了更方便地处理头节点为空的边界条件
4. 遍历结束后，我们将 \textit{large}large 的 \textit{next}next 指针置空，这是因为当前节点复用的是原链表的节点，而其 \textit{next}next 指针可能指向一个小于 x的节点

解题代码

var partition = function(head, x) {
    let small = new ListNode(0);
    const smallHead = small;
    let large = new ListNode(0);
    const largeHead = large;
    while (head !== null) {             //边界处理
        if (head.val < x) {
            small.next = head;
            small = small.next;   //small 的next 指针指向该节点，
        } else {
            large.next = head;
            large = large.next;   //否则将large 的 next 指针指向该节点。
        }
        head = head.next;
    }
    large.next = null;
    small.next = largeHead.next;
    return smallHead.next;
};