725. 分隔链表

### 解题思路

1.首先求出链表中一共有多少个元素n

2.然后通过n/k判断一共要分成几块

3.通过n%k判断多出来多少个节点,把他们分别分配到对应的块中去

### 代码

class Solution {
public:
    vector<ListNode*> splitListToParts(ListNode* root, int k) {
         int n = 0;
         for(auto p=root;p;p=p->next) n++;  //统计元素个数

         int part = n / k;
         int remainder = n % k;
         vector<ListNode*> res(k,NULL);  
         ListNode* p = root;
         if(n < k) k = n;
         for(int i = 0; i < k; ++i){
            int tempLen = remainder > 0 ? part+1:part;  //根据余数的数量判断当前块要不要多开一个位置给多余的节点
            ListNode* nhead = p;
            for(int j = 0; j < tempLen-1; ++j) p = p->next;
            auto c = p;
            p = p->next;
            c->next = NULL;
            res[i] = nhead;
            if(remainder) remainder--;
         }

         return res;
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,我理解了您的问题。下面是C++代码实现: ```c++ #include <iostream> #include <vector> using namespace std; struct ListNode { int val; ListNode *next; ListNode(int x) : val(x), next(NULL) {} }; vector<ListNode*> splitListToParts(ListNode* head, int k) { vector<ListNode*> res(k, nullptr); int len = 0; for (ListNode* p = head; p; p = p->next) ++len; int n = len / k, r = len % k; ListNode* p = head, *prev = nullptr; for (int i = 0; p && i < k; ++i, --r) { res[i] = p; for (int j = 0; j < n + (r > 0); ++j) { prev = p; p = p->next; } prev->next = nullptr; } return res; } int main() { // 创建链表 ListNode* head = new ListNode(1); head->next = new ListNode(2); head->next->next = new ListNode(3); head->next->next->next = new ListNode(4); head->next->next->next->next = new ListNode(5); // 分隔链表 vector<ListNode*> parts = splitListToParts(head, 3); for (int i = 0; i < parts.size(); ++i) { cout << "Part " << i + 1 << ": "; ListNode* p = parts[i]; while (p) { cout << p->val << " "; p = p->next; } cout << endl; } return 0; } ``` 这段代码中,我们使用了 `vector` 来存储分隔后的链表。首先计算链表长度 `len`,然后计算每个部分的长度 `n = len / k` 和余数 `r = len % k`。接下来我们遍历链表,将每个部分的头结点存储到 `vector` 中,然后将链表断开,重复上述步骤,最后返回 `vector` 即可。 运行结果如下: ``` Part 1: 1 2 Part 2: 3 4 Part 3: 5 ``` 这里我们将长度尽可能相等的策略应用到了每个部分中,因此部分 1 和 2 的长度相等,部分 3 的长度比它们短 1。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值