[路飞]_725. 分隔链表

「这是我参与2022首次更文挑战的第16天，活动详情查看：2022首次更文挑战」

725. 分隔链表

给你一个头结点为 head 的单链表和一个整数 k ，请你设计一个算法将链表分隔为 k 个连续的部分。

每部分的长度应该尽可能的相等：任意两部分的长度差距不能超过 1 。这可能会导致有些部分为 null 。

这 k 个部分应该按照在链表中出现的顺序排列，并且排在前面的部分的长度应该大于或等于排在后面的长度。

返回一个由上述 k 部分组成的数组。

「示例1:」

输入：head = [1,2,3], k = 5
​
输出：[[1],[2],[3],[],[]]
​
解释：
第一个元素 output[0] 为 output[0].val = 1 ，output[0].next = null 。
最后一个元素 output[4] 为 null ，但它作为 ListNode 的字符串表示是 [] 。

「示例2:」

输入：head = [1,2,3,4,5,6,7,8,9,10], k = 3
​
输出：[[1,2,3,4],[5,6,7],[8,9,10]]
​
解释：
输入被分成了几个连续的部分，并且每部分的长度相差不超过 1 。前面部分的长度大于等于后面部分的长度。

「提示:」

1. 链表中节点的数目在范围 [0, 1000]
2. 0 <= Node.val <= 1000
3. 1 <= k <= 50

​

解题思路

​

1. 根据题意，我们先复制这个链表，并且记录链表有几个节点
2. 把链表分成k个部分，并且把链表长度向k取余
3. new一个k长度的数组，默认值为null
4. 循环链表，分别向新数组里面添加值
5. 最后把这个数组里面的值添加指向，形成链表并且返回即可

​

代码实现

​

/**
 * 725. 分隔链表
 * @param {ListNode} head
 * @param {number} k
 * @return {ListNode[]}
 */
var splitListToParts = function (head, k) {
  let n = 0;
  let temp = head;
  while (temp) {
    n++;
    temp = temp.next;
  }
​
  let parts = Math.floor(n / k); // 0
  let residue = n % k; // 3
​
  let res = new Array(k).fill(null);
  let cur = head;
  for (let i = 0; i < k && cur; i++) {
    res[i] = cur;
    let partSize = parts + (i < residue ? 1 : 0);
    for (let j = 1; j < partSize; j++) {
      cur = cur.next;
    }
    let p = cur.next;
    cur.next = null;
    cur = p;
  }
  return res;
};

如果你对这道题目还有疑问的话，可以在评论区进行留言；