0-13每日一题：Leetcode725. 分隔链表

最新推荐文章于 2024-09-29 13:57:06 发布

原创最新推荐文章于 2024-09-29 13:57:06 发布 · 161 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#链表

每日一题专栏收录该内容

144 篇文章

订阅专栏

本文介绍了解决LeetCode725题目的方法：分隔链表。通过计算链表长度并均匀分割，确保每个部分尽可能等长。文章详细展示了如何遍历链表、分割链表，并保留每个部分的头指针。

Leetcode725. 分隔链表

计算出每一段链表长度，分隔链表，分隔的时候，记得存下一个链表头指针防止丢失

/**
 * 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*> res;
        int cnt = 0;
        auto p = head, next = p;
        while (p) p = p->next, cnt ++ ;
        int a = cnt / k, b = cnt % k;
        p = head;
        while (p) {
            res.push_back(p);
            for (int i = 0; i < a - 1 + (b != 0); i ++ ) p = p->next;
            if (b) {
                b -- ;
            }
            next = p->next;
            p->next = nullptr;
            p = next;
            k --;
        }
        while (k -- ) res.push_back(nullptr);
        return res;
    }
};