725. 分隔链表
Posted hequnwang10
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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 。前面部分的长度大于等于后面部分的长度。
二、解题
要求分割的每一段的长度差不能超过1。所以每段长度最基本的长度为链表的长度cnt/k,取整,然后余数为cnt%k,将余数分别分配给基数一个数值即可,知道余数分配完为止。这样就可以将链表分割。
/**
* 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)
//统计链表的长度,然后根据K个连续的部分算出每个部分至少需要几个数,然后根据余数cnt%k每次向连续部分+1,即可。
int cnt = 0;
ListNode cur = head;
while(cur != null)
cnt++;
cur = cur.next;
//定义返回数组
ListNode[] res = new ListNode[k];
int sum = cnt/k;
int rem = cnt%k;
cur = head;
while(cur!=null)
//每段数组的长度,为sum+1(这里的1来自于余数分配)
for(int i = 0;cur!=null && i<k;i++)
res[i] = cur;
int size = sum+(rem-- > 0?1:0);
//分配数组段
for(int j = 0;j<size-1;j++)
cur = cur.next;
//分割数组
ListNode pre = cur.next;
cur.next = null;
cur = pre;
return res;
以上是关于725. 分隔链表的主要内容,如果未能解决你的问题,请参考以下文章