725. 分隔鏈表

給你一個頭結(jié)點為 head 的單鏈表和一個整數(shù) k ，請你設(shè)計一個算法將鏈表分隔為 k 個連續(xù)的部分。

每部分的長度應(yīng)該盡可能的相等：任意兩部分的長度差距不能超過 1 。這可能會導(dǎo)致有些部分為 null 。

這 k 個部分應(yīng)該按照在鏈表中出現(xiàn)的順序排列，并且排在前面的部分的長度應(yīng)該大于或等于排在后面的長度。

返回一個由上述 k 部分組成的數(shù)組。

示例 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]]
解釋：
輸入被分成了幾個連續(xù)的部分，并且每部分的長度相差不超過 1 。前面部分的長度大于等于后面部分的長度。

提示：

• 鏈表中節(jié)點的數(shù)目在范圍 [0, 1000]
• 0 <= Node.val <= 1000
• 1 <= k <= 50

解題思路

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

代碼

/*** 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;} }

總結(jié)

以上是生活随笔為你收集整理的725. 分隔链表的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。