分隔链表

2021-03-17

题目描述

给你一个链表的头节点 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]

解题思路

解题思路：创建两个链表，一个用于插入小于x的元素，一个用于插入大于x的元素，最后把两个链表合并即可

思路：
1.这道题类似于快排，找到一个中间值，比它大的放后面，比它小的放后面，但是不同的是，分割链表的相对位置要保持不变
2.创建两个链表，一个存储比x小的元素，另一个是比x大的元素
3.为两个链表定义两个指针
4.定义原链表的头指针，然后进行比较，连接到对应的链表，然后进行移动
5.将两个链表拼接到了一起

const partition = function (head, x) {
    if (head === null) {
        return null;
    }
    //创建两个链表l1,l2.p1和p2分别指向l1和l2链表的最后一个元素的位置.p为原链表上的指针，从头开始移动,q要保存p链表的下一个元素的地址.
    let l1 = new ListNode(), l2 = new ListNode(), p1 = l1, p2 = l2, p = head, q;
    while (p) {
        q = p.next;//q指向p.next
        p.next = null;
        if (p.val < x) {
            //如果p的值小于x，插入到p1的后边,并且p1 = p（p1指向最后一个元素）
            p1.next = p;
            p1 = p;
        } else {
            p2.next = p;
            p2 = p;
        }
        p = q;//p指针向后移动
    }
    p1.next = l2.next;//l1链表的最后一个元素位置是p1，把两个链表合并的话只要把p1.next指向l2链表的第一个元素的位置
    return l1.next;//返回l1链表的第一个元素的位置
}