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