首先直接进入主题,题目链接🔗力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台
源代码在最后,有更优解的朋友欢迎在评论里指导我一番!
1.题目分析
通过题目分析得出结论:
1. 将链表分为k个子链表
2. 用一个数组存放这k个子链表,数组的长度就是k
3. 任意两个子链表的长度差不能超过1,也就是要么子链表长度都是2或者别的数字,要么子链表之间可以是2,2,2,1或者3,3,2等。子链表的长度跟链表的长度和k有关。
4. 子链表存放到数组的顺序不能改变,也就是说链表是从头开始分隔的,依次再存放到数组里。
5. 既然数组是存放子链表的,那准确来说数组元素是链表某一节点的地址,之后通过这个节点能找到后面的节点,直到每个子链表的节点到NULL的时候。
我们用图解来展示一下:
我们为了更形象的展现数组里的元素,直接把子链表放入,但我们要知道存放的其实是每个子链表的头节点的地址!
知道这样之后,我们直接上代码,代码里有详细介绍!
2.源代码
struct ListNode** Create(int k,int* returnSize)
{
*returnSize = k;
struct ListNode** output = (struct ListNode**)malloc(sizeof(struct ListNode*) * k);
if(output == NULL)
{
perror("malloc fail");
exit(-1);
}
//注意!!!!!!1
//这里不初始化会报错的,因为我们这个指针数组存的是指针,必须初始化为空,否则都是野指针
//你想想一个指针数组存放的都是野指针,没有任何意义!
for(int i = 0; i < k; i++)
{
output[i] = NULL;
}
return output;
}
struct ListNode** splitListToParts(struct ListNode* head, int k, int* returnSize)
{
//我们要知道这个题返回的是什么,本题返回的是一个二级指针,也就是存放一级指针的地址,所以在这里开辟的数组
//应该是存放的是链表节点的地址
//我们清楚返回值之后就写题吧
//创建一个指针数组output,用来存放链表节点的地址
struct ListNode** output = Create(k,returnSize);
//链表为空的时候,这个数组都是空指针,因为我们初始化为空了,所以直接返回就可以了。
if(head == NULL)
return output;
//链表不为空的时候
//我们先求链表长度,看分隔的段数是否大于链表长度
struct ListNode* cur = head;
int len = 0;
while(cur)
{
len++;
cur = cur->next;
}
//当链表长度小于要分隔的段数时,很明显链表的节点是不够用的
//当添加完节点之后,剩下的用空指针补齐,我们上面初始化了已经,所以不需要对空指针这里做代码
if(len < k)
{
cur = head;
int i = 0;
while(cur)
{
struct ListNode* next = cur->next;
output[i] = cur;
cur->next = NULL;
cur = next;
i++;
}
}
//当链表长度大于分隔的段数k
//我们这里就要注意一点,任意两部分长度的差距不能超过一
else
{
//这里我们通过n来知道分k段,剩了几个节点
int n = len % k;
//通过m知道,最开始一段分几个节点。
int m = len / k;
int i = 0;
cur = head;
//任意两个部分的长度差距不超过一,那么我们就让第一部分加1,第二部分加1,依次加一,但是加几个呢?
//看n是几,就加几次
while(n--)
{
m = len / k;
output[i] = cur;
while(m)
{
cur = cur->next;
m--;
}
struct ListNode* next = cur->next;
cur->next = NULL;
cur = next;
i++;
}
//在分配完剩余节点个数之后,就正常的一段应该是几个节点
//这里的条件就是下标i小于数组元素个数
while(i < k)
{
m = len / k - 1;
output[i] = cur;
while(m)
{
cur = cur->next;
m--;
}
struct ListNode* next = cur->next;
cur->next = NULL;
cur = next;
i++;
}
}
//最后返回output
return output;
}