LeetCode 86 分隔链表

参加了 kkb 的门徒培训，想要记录下自己的成长过程。每天一篇，坚持不断更 --- 2022.3.12

题目描述

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

提示：

• 链表中节点的数目在范围 [0, 200] 内
• -100 <= Node.val <= 100
• -200 <= x <= 200

解题思路

拆分链表再组合

/**
 * @param {ListNode} head
 * @param {number} x
 * @return {ListNode}
 */
var partition = function (head, x) {
  // 判断 head 是否为空
  if (!head || !head.next) return head;

  // 用两个额外的链表保存
  const r1 = new ListNode(), r2 = new ListNode();
  let p1 = r1, p2 = r2, p = head;

  // 遍历链表按题要求分配到两个链表中
  while (p) {
    if (p.val < x) {
      p1.next = p;
      p1 = p;
    } else {
      p2.next = p;
      p2 = p;
    }
    p = p.next;
  }

  // 将两个链表首尾连接起来
  p1.next = r2.next;
  // 将尾部设置为空
  p2.next = null;

  return r1.next;
};