1. 题目描述
给定一个头结点为 root
的链表, 编写一个函数以将链表分隔为 k
个连续的部分。
每部分的长度应该尽可能的相等: 任意两部分的长度差距不能超过 1,也就是说可能有些部分为 null。
这k个部分应该按照在链表中出现的顺序进行输出,并且排在前面的部分的长度应该大于或等于后面的长度。
返回一个符合上述规则的链表的列表。
举例: 1->2->3->4, k = 5 // 5 结果 [ [1], [2], [3], [4], null ]
示例 1:**
输入:
root = [1, 2, 3], k = 5
输出: [[1],[2],[3],[],[]]
解释:
输入输出各部分都应该是链表,而不是数组。
例如, 输入的结点 root 的 val= 1, root.next.val = 2, \root.next.next.val = 3, 且 root.next.next.next = null。
第一个输出 output[0] 是 output[0].val = 1, output[0].next = null。
最后一个元素 output[4] 为 null, 它代表了最后一个部分为空链表。
示例 2:
输入:
root = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], k = 3
输出: [[1, 2, 3, 4], [5, 6, 7], [8, 9, 10]]
解释:
输入被分成了几个连续的部分,并且每部分的长度相差不超过1.前面部分的长度大于等于后面部分的长度。
提示:
root
的长度范围:[0, 1000]
.- 输入的每个节点的大小范围:
[0, 999]
. k
的取值范围:[1, 50]
.2. 解题思路
对于这道题目,我们可以先计算链表的长度length,尽可能均分链表。然后再计算每个子链表的长度:先计算平均每个子链表的长度为length/k,倘若不能整除,剩余的length%k个元素,则前length%k个子链表的长度再加一。
复杂度分析:
- 时间复杂度:O(N + k),N 指的是链表的结点数,若 k 很大,则还需要添加许多空列表。
-
3. 代码实现
/**
* Definition for singly-linked list.
* function ListNode(val) {
* this.val = val;
* this.next = null;
* }
*/
/**
* @param {ListNode} root
* @param {number} k
* @return {ListNode[]}
*/
var splitListToParts = function(root, k) {
let cur = root
let length = 0
while(cur){
cur = cur.next
length++
}
// 每一项的长度
let width = Math.floor(length / k), curry = length % k
let num, temp
let res = new Array(k);
for(let i = 0; i < k; i++) {
// 计算从剩余链表中截取的长度
num = i < curry ? width + 1 : width
// 从剩余链表中截取num个元素
temp = cur = root;
for(let j = 1; cur && j < num; j++) {
cur = cur.next;
}
// 修改root指向
if(cur) {
root = cur.next;
cur.next = null;
} else {
root = null;
}
res[i] = temp;
}
return res
};
4. 提交结果