题目介绍
给你一个头结点为 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 。前面部分的长度大于等于后面部分的长度。
提示:
- 链表中节点的数目在范围
[0, 1000] 0 <= Node.val <= 10001 <= k <= 50leetcode-725 分隔链表
b站视频
解题思路
题目要求将链表分成 k 份,每一份的长度尽量相等,即使不相等,任意两部分的长度差距不能超过 1,并且排在前面部分的长度应该大于或等于排在后面的长度
可以先将链表的长度除以 k 份,得到结果的整数部分即为每部分的最小长度,如果余数不为 0,将余数平均分配到前面的几个部分长度中,这样便能达到题目的要求
解题步骤:
- 遍历链表,计算链表的长度
- 链表长度 除以 要分成 k 份,取整数部分为每一份的最小长度
- 计算 链表长度 除以 要分成 k 份 的余数,平均分配到前面的 n 份中,即前面 n 份的长度比后面的长度多 1
- 根据每一份的长度进行链表的分隔
解题代码
var splitListToParts = function(head, k) {
let n = 0, p = head
const res = []
// 遍历链表,获取链表长度
while (p) {
n++
p = p.next
}
// 获取每一份的最小长度
var num1 = Math.floor(n / k)
// 前面 num2 份长度多 1
var num2 = n % k
p = head
for(let i = 0; i < k; i++) {
res.push(p)
// 如果 p 已经为空,当前部分的长度必定为 0,不需要继续后面的步骤
if (!p) continue
// 前 num2 部分长度+1
let num3 = i < num2 ? num1 + 1 : num1
while (--num3) {
p = p.next
}
// 将每一部分的尾节点截断,防止长度超过规定
const temp = p.next
p.next = null
p = temp
}
return res
};