725. 分隔鏈表

給你一個頭結點為 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 <= 1000
• 1 <= k <= 50

解題思路

統計出節點的個數，將節點分為k份，首先使用總數/k得出每一個子鏈表至少需要多少個節點，然后再計算出多出來的節點個數gap，將這些節點均攤到前gap個子鏈表，每個子鏈表增加一個節點

代碼

/*** Definition for singly-linked list.* public class ListNode {*     int val;*     ListNode next;*     ListNode() {}*     ListNode(int val) { this.val = val; }*     ListNode(int val, ListNode next) { this.val = val; this.next = next; }* }*/
class Solution {public ListNode[] splitListToParts(ListNode head, int k) {int cnt=0;ListNode cntHead=head;while(cntHead!=null){cntHead=cntHead.next;cnt++;}int d=cnt/k,gap=cnt-d*k,idx=0;ListNode[] res=new ListNode[k];while(head!=null){ListNode nh=head;res[idx++]=nh;int num=d;if(gap>0){num++;gap--;}while(nh!=null&&num>1){nh=nh.next;num--;   }if(nh==null) return res;head=nh.next;nh.next=null;}return res;}
}