题目
给你一个链表的头节点 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;
};