力扣小白刷题之725题分隔链表

题目描述

给定一个头结点为 root 的链表，编写一个函数以将链表分隔为 k 个连续的部分。
每部分的长度应该尽可能的相等：任意两部分的长度差距不能超过1，也就是说可能有些部分为 null。
这 k 个部分应该按照在链表中出现的顺序进行输出，并且排在前面的部分的长度应该大于或者等于后面的长度。
返回一个符和上述规则的链表的列表。

输入输出各部分都应该是链表。

思路

参考自：https://leetcode-cn.com/problems/split-linked-list-in-parts/solution/qiao-miao-lei-bi-si-lu-qing-xi-dai-ma-jian-dan-er-/

本题实质：考察如何划分一个整数 n
巧妙类比：

• 先数出有多少块小糖，得 n 块；
• 将 n 块小糖分给 k 个小朋友，从左往右均分，每人(n / k) 块；
• 将剩下的(n % k)块小糖再从左往右一人一块，直至分完。

解题步骤：
参考自：https://leetcode-cn.com/problems/split-linked-list-in-parts/solution/python3-zhu-xing-jie-shi-fen-ge-lian-biao-by-han-h/

• 首先遍历链表得到链表长度 N，要把 N 分为 k 段；
• 总长度除以 k 就得到每一段应分配的基础长度：size = N / k;
• 题目要求排在前面的部分的长度应该大于或等于后面的长度，所以总长度除以 k 得到的余数 mod 就是前 mod 段每段的长度为 size + 1;
• 接下来就需要两层循环：
  • 外层循环遍历 k 段，每次都把当前的节点加入结果列表；
  • 内层循环遍历每段的大小，并后移指针：cur = cur.next，直到到达该段应该分配的长度后，就把这一段截断，后面的链表会在之后的循环中继续分配。
  • 特别注意内层循环的前提条件是当前结点 cur 必须存在，如果 cur 为 null，cur.next 不存在就会报错。

代码