1. 题目描述

给定一个头结点为 root 的链表, 编写一个函数以将链表分隔为 k 个连续的部分。

每部分的长度应该尽可能的相等: 任意两部分的长度差距不能超过 1,也就是说可能有些部分为 null。

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

返回一个符合上述规则的链表的列表。

举例: 1->2->3->4, k = 5 // 5 结果 [ [1], [2], [3], [4], null ]

示例 1:**

  1. 输入:
  2. root = [1, 2, 3], k = 5
  3. 输出: [[1],[2],[3],[],[]]
  4. 解释:
  5. 输入输出各部分都应该是链表,而不是数组。
  6. 例如, 输入的结点 root val= 1, root.next.val = 2, \root.next.next.val = 3, root.next.next.next = null
  7. 第一个输出 output[0] output[0].val = 1, output[0].next = null
  8. 最后一个元素 output[4] null, 它代表了最后一个部分为空链表。

示例 2:

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

提示:

  • root 的长度范围: [0, 1000].
  • 输入的每个节点的大小范围:[0, 999].
  • k 的取值范围: [1, 50].

    2. 解题思路

    对于这道题目,我们可以先计算链表的长度length,尽可能均分链表。然后再计算每个子链表的长度:先计算平均每个子链表的长度为length/k,倘若不能整除,剩余的length%k个元素,则前length%k个子链表的长度再加一。

复杂度分析:

  • 时间复杂度:O(N + k),N 指的是链表的结点数,若 k 很大,则还需要添加许多空列表。
  • 空间复杂度:O(max(N, k)),存放答案所需的空间。

    3. 代码实现

    1. /**
    2. * Definition for singly-linked list.
    3. * function ListNode(val) {
    4. * this.val = val;
    5. * this.next = null;
    6. * }
    7. */
    8. /**
    9. * @param {ListNode} root
    10. * @param {number} k
    11. * @return {ListNode[]}
    12. */
    13. var splitListToParts = function(root, k) {
    14. let cur = root
    15. let length = 0
    16. while(cur){
    17. cur = cur.next
    18. length++
    19. }
    20. // 每一项的长度
    21. let width = Math.floor(length / k), curry = length % k
    22. let num, temp
    23. let res = new Array(k);
    24. for(let i = 0; i < k; i++) {
    25. // 计算从剩余链表中截取的长度
    26. num = i < curry ? width + 1 : width
    27. // 从剩余链表中截取num个元素
    28. temp = cur = root;
    29. for(let j = 1; cur && j < num; j++) {
    30. cur = cur.next;
    31. }
    32. // 修改root指向
    33. if(cur) {
    34. root = cur.next;
    35. cur.next = null;
    36. } else {
    37. root = null;
    38. }
    39. res[i] = temp;
    40. }
    41. return res
    42. };

    4. 提交结果

    image.png