LeetCode 725 分隔链表[链表 双指针] HERODING的LeetCode之路

Posted 2021-10-11 HERODING23

解题思路：
首先获取整个链表的长度，接着获取平均的长度以及余数remain，余数remain代表着前remain个子链表的长度为avg + 1，这样才能满足排在前面的部分的长度应该大于或等于后面的长度且任意两部分的长度差距不能超过 1的条件，在断链环节，定义前后指针，一个是作为头节点，一个是作为尾部进行断链，当满足条件的时候，尾部指针直接指向nullptr，代码如下：

/**
 * 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) {
        int len = 0;
        ListNode* temp = head;
        while(temp) {
            len ++;
            temp = temp -> next;
        }
        // 定义平均长度和余数
        int avg = len / k;
        int remain = len % k;
        vector<ListNode *> res(k, nullptr);
        // 定义前后指针
        ListNode* node = head;
        ListNode* pre = nullptr;
        for(int i = 0; i < k; i ++) {
            res[i] = node;
            int length = remain > 0 ? avg + 1 : avg;
            for(int j = 0; j < length; j ++) {
                pre = node;
                node = node -> next;
            }
            // 断开子链
            if(pre) pre -> next = nullptr;
            // 更新剩余余数个数
            if(remain) remain --;
        }
        return res;
    }
};


/*作者：heroding
链接：https://leetcode-cn.com/problems/split-linked-list-in-parts/solution/cshuang-zhi-zhen-by-heroding-1h1k/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/