leetcode #725. 分隔链表

题目：

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

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

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

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

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

思路：

1、先遍历链表，计算长度n
2、用n // k，计算出n1 为所需的基础片段长度
3、用ｎ％ k，计算出需要+1 的片段个数
4、用两个循环：外循环遍历k遍，每次都把当前的节点加入结果列表；
5、内层循环遍历每段的大小，并后移指针：cur = cur.next，直到到达该段应分配的长度后，就把这一段截断，后面的链表会在之后的循环中继续分配。特别注意内层循环的前提条件是当前节点 cur 必须存在，如果 cur 为 Null，cur.next 不存在就会报错。

代码：

class Solution:
    def splitListToParts(self, root: ListNode, k: int) -> List[ListNode]:
        p = root
        n = 0
        while p:
            n += 1
            p = p.next 
        n1 = n // k # 基础长度
        n2 = n % k  #前n2个链表需要在基础长度上+1

        cur = root 
        ans = []
        for i in range(k): #大循环
            ans.append(cur)
            size = n1 + (1 if n2>0 else 0) # 长片段的长度
            if cur : #判断cur是否存在
                for k in range(size-1): #已经放了一个node进去，所以要 -1
                    cur = cur.next 
                n2 -= 1 
                tmp = cur.next #维护cur的下一个结点
                cur.next = None #当作这个片段的结尾
                cur = tmp
        return ans