86. 分隔链表(JS实现)
86. 分隔链表(JS实现)
1 题目
给定一个链表和一个特定值 x,对链表进行分隔,使得所有小于 x 的节点都在大于或等于 x 的节点之前。
你应当保留两个分区中每个节点的初始相对位置。
示例:
输入: head = 1->4->3->2->5->2, x = 3
输出: 1->2->2->4->3->5
2 思路
这道题思路主要是新建两个列表,将大于或等于目标的元素、以及小于目标的元素分开,然后再将两个列表连接起来
3代码
/**
* Definition for singly-linked list.
* function ListNode(val) {
* this.val = val;
* this.next = null;
* }
*/
/**
* @param {ListNode} head
* @param {number} x
* @return {ListNode}
*/
var partition = function(head, x) {
if (!head || !head.next) return head;
let smallLink; //小于target的元素列表头
let bigLink; //比大于或等于target的元素列表头
let bigHead;
let prev;
while(head) {
if (head.val >= x) {
if (!bigLink) {
bigLink = head;
} else {
bigHead.next = head;
}
bigHead = head;
} else {
if (prev) {
prev.next = head;
} else {
smallLink = head;
}
prev = head;
}
head = head.next;
}
if (!prev) return bigLink; //全部元素都大于或相等target
if (!bigLink) return smallLink; //全部元素都小于target
prev.next = bigLink; //将大于列表接在小于列表后面
bigHead.next = null;
return smallLink;
};