题源: 分隔链表
算法流程:
- 遍历链表,求出链表长度
listLen listLen除以k得到平均长度splitLen和余数remainder。- 题目要求
- 每部分的长度应该尽可能的相等:
- 前面的部分的长度应该大于或等于后面的长度
- 因此
remainder应当在前remainder上平均分配- 举个例子:
root = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11], k = 3,len = 11, k = 3, splitLen, remainder = 11 // 3, 11 % 3 = 3, 2- 最后划分的链表:
[1, 2, 3, 4], [5, 6, 7, 8], [ 9, 10, 11]
- 举个例子:
- 剩下的就是分隔链表了
具体代码
def splitListToParts(head, k):
res, listLen, tail = [None] * k, 0, head
while tail != None:
tail = tail.next
listLen += 1
splitLen, remainder = listLen // k, listLen % k
tail, idx = head, 0
while tail:
res[idx] = tail
qlast = None
for i in range(splitLen + (remainder > idx)):
qlast, tail= tail, tail.next
idx += 1
if qlast:
qlast.next = None
return res
复杂度分析:
- 时间复杂度:
O(n),n为链表长度。 - 空间复杂度:
O(1),返回值不计入空间复杂度,只借用了常量空间

297

被折叠的 条评论
为什么被折叠?



