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;
};