力扣：725 分隔链表

Posted 2021-10-10 战场小包

题源: 分隔链表

算法流程：

1. 遍历链表，求出链表长度listLen
2. listLen除以k得到平均长度splitLen和余数remainder。
3. 题目要求
   1. 每部分的长度应该尽可能的相等:
   2. 前面的部分的长度应该大于或等于后面的长度
4. 因此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]
5. 剩下的就是分隔链表了

具体代码

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)，返回值不计入空间复杂度，只借用了常量空间