LeetCode 725. 分隔链表 | C语言版

NowCoder Offer 725. 分隔链表

题目描述

题目地址：725. 分隔链表
给你一个头结点为 head 的单链表和一个整数 k ，请你设计一个算法将链表分隔为 k 个连续的部分。

每部分的长度应该尽可能的相等：任意两部分的长度差距不能超过 1 。这可能会导致有些部分为 null 。

这 k 个部分应该按照在链表中出现的顺序排列，并且排在前面的部分的长度应该大于或等于排在后面的长度。

返回一个由上述 k 部分组成的数组。

解题思路

思路一：使用

代码实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */


/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
     /*
    *把一个链表分割成K个连续的部分，要求任意两部分的长度差距不能超过1，前面的部分的长度应该大于等于后面部分的长度，有些部分可能为NULL。
    [就是类比给n个孩子分k个苹果，先平均分，然后剩下的苹果从前往后一个个给]
    */

struct ListNode** splitListToParts(struct ListNode* head, int k, int* returnSize){
    //申请内存(一段连续二维指针，可写成数组的格式，用下标访问，output[i]对应的是第i个链表的表头)
    struct ListNode** output=(struct ListNode**)malloc(k*sizeof(struct ListNode*));
    //链表长度
    int length=0;
    //当前节点cur
    struct ListNode* cur=head;

    while(cur!=NULL){
        length++;
        cur=cur->next;
    }

    //总体思路:类比给k个孩子分N个苹果，先平均分，再把多的从头往后一个个分
    //遍历完后把链表重置到链表开头
    cur=head;
    //均分后余下的节点数
    int mod=length%k;
    //1.先均分：每个部分本应有的结点数
    int size=length/k;

    for(int i=0;i<k;i++){
        output[i]=cur;
        if(cur!=NULL){
            //2.再把多的从头往后一个个分 curSize将余下的节点分配到每部分链表
            int curSize=size+(mod-- >0 ? 1:0);
            //分割链表
            for(int j=0;j<curSize-1;j++) cur=cur->next;
            struct ListNode* next=cur->next;
            cur->next=NULL;
            cur=next;
        }
    }
    *returnSize=k;
    return output;
}

运行结果

参考文章：

https://leetcode.cn/problems/split-linked-list-in-parts/solutions/263735/fen-ge-lian-biao-cyu-yan-by-nuclear-3/?orderBy=most_votes&languageTags=cpp%2Cc

思路二：减少遍历节点数

代码实现

在这里插入代码片

运行结果

参考文章：