LeetCode刷题之分隔链表(图解➕代码)

        首先直接进入主题,题目链接力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台

源代码在最后,有更优解的朋友欢迎在评论里指导我一番!

1.题目分析

LeetCode刷题之分隔链表(图解➕代码)_第1张图片

通过题目分析得出结论:

        1. 将链表分为k个子链表

        2. 用一个数组存放这k个子链表,数组的长度就是k

        3. 任意两个子链表的长度差不能超过1,也就是要么子链表长度都是2或者别的数字,要么子链表之间可以是2,2,2,1或者3,3,2等。子链表的长度跟链表的长度和k有关

        4. 子链表存放到数组的顺序不能改变,也就是说链表是从头开始分隔的,依次再存放到数组里。

        5. 既然数组是存放子链表的,那准确来说数组元素是链表某一节点的地址,之后通过这个节点能找到后面的节点,直到每个子链表的节点到NULL的时候。

我们用图解来展示一下:

        我们为了更形象的展现数组里的元素,直接把子链表放入,但我们要知道存放的其实是每个子链表的头节点的地址!

LeetCode刷题之分隔链表(图解➕代码)_第2张图片

知道这样之后,我们直接上代码,代码里有详细介绍!

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

你可能感兴趣的:(leetcode,链表,算法)