leetCode: 86. 分隔链表

91 阅读1分钟

题目

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

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

例如:

输入: head = [1,4,3,2,5,2], x = 3
输出:[1,2,2,4,3,5]

解题思路

需要维护两个虚拟列表,一个small和一个large链表分别储存小于x和大于等于x的值,遍历链表,分别取出放入两个链表中,遍历结束后将small链表的next节点与large的头节点相连即可

var partition = function(head, x) {
    // 定义大小虚拟头几点
    let vSmall = new ListNode(0), vLarge = new ListNode(0);
    // 定义虚拟节点的头节点,两链表相连时用到
    const smallHead = vSmall, vLargeHead = vLarge;
    // 循环遍历链表分成大小两个链表
    while(head) {
        // 小于x的值放入小链表,大于x的值放入大链表
        if (head.val < x) {
            vSmall.next = head;
            vSmall = vSmall.next;
        } else {
            vLarge.next = head;
            vLarge = vLarge.next;
        }
        // 进入下一次循环
        head = head.next;
    }
    
    // 大小链表相连:大链表末尾置空,小链表末尾next节点指向大链表头节点,最后返回小链表
    vLarge.next = null;
    vSmall.next = vLargeHead.next;
    return smallHead.next;
};