725. 分隔链表
Posted 我要出家当道士
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了725. 分隔链表相关的知识,希望对你有一定的参考价值。
目录
1、Question
给你一个头结点为 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来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/split-linked-list-in-parts
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2、Analysis
思路很简单,先按照平均数(向下取整)给每个子链表分配容量,再将余数平分给前几个子链表,每个链表分一个,分完为止。这题的难点在于如何减少链表的遍历次数,降低时间复杂度,思路也很简单,空间换时间。下面代码,只需要遍历链表一次。
3、Code
struct ListNode** splitListToParts(struct ListNode* head, int k, int* returnSize){
#define NEW (struct ListNode *)malloc(sizeof(struct ListNode))
int linkLen = 0;
struct ListNode* tmp = head;
while(tmp != NULL)
{
linkLen++;
tmp = tmp->next;
}
int ave = linkLen / k;
int mod = linkLen % k;
int cnt = 0;
int index = -1;
struct ListNode** res = (struct ListNode **)malloc(sizeof(struct ListNode*) * k);
struct ListNode** rec = (struct ListNode **)malloc(sizeof(struct ListNode*) * k);
for (int i = 0; i < k; i++)
res[i] = NULL;
while(head != NULL)
{
if (!cnt)
{
cnt = ave;
if (mod)
{
mod--;
cnt++;
}
index++;
res[index] = NEW;
res[index]->val = head->val;
res[index]->next = NULL;
rec[index] = res[index];
}
else
{
struct ListNode* node = NEW;
node->val = head->val;
node->next = NULL;
tmp = rec[index];
tmp->next = node;
rec[index] = node;
}
head = head->next;
cnt--;
}
*returnSize = k;
return res;
}
4、Execution
以上是关于725. 分隔链表的主要内容,如果未能解决你的问题,请参考以下文章
[M链表] lc725. 分隔链表(模拟+代码优化+代码实现)