725. 分隔链表

最新推荐文章于 2024-08-31 00:26:34 发布

cx_cs

最新推荐文章于 2024-08-31 00:26:34 发布

阅读量180

点赞数

分类专栏：链表文章标签： leetcode 算法链表

本文链接：https://blog.csdn.net/cx_cs/article/details/121469506

版权

链表专栏收录该内容

24 篇文章 0 订阅

订阅专栏

该博客主要讨论如何将一个单链表平均分成k个连续的部分。提供的C++代码实现了一个函数，根据输入的链表头节点和k值，返回一个包含k个链表部分的向量。算法考虑了两种情况：当链表长度不能被k整除时，会将多余的节点逐个分配到前面的子链表中。

摘要由CSDN通过智能技术生成

要求：分成k份
思路：两种情况，不够分每次一个，够分把余数一个个加上

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    vector<ListNode*> splitListToParts(ListNode* head, int k) {
        vector<ListNode*> ans(k);
        ListNode *p=head;
        int cnt=0;
        while(p){
            p=p->next;
            ++cnt;
        }
        int a=cnt/k,b=cnt%k;
        ListNode *cur=head,*tmp;
        for(int i=0;i<k;++i){
            ans[i]=cur;
            int len;
            if(a==0)len=0;//不够分直接加一个
            else len=i<b?a:a-1;//够分的话把余数一个个加上
            while(len-->0&&cur)cur=cur->next;
            if(cur){
                tmp=cur->next;
                cur->next=nullptr;
                cur=tmp;
            }
        }
        return ans;
    }
};