力扣 725. 分隔链表 链表+思维

https://leetcode-cn.com/problems/split-linked-list-in-parts/

思路：假设链表长度为$n$，要将其分成$k$部分，我们可以先使得每部分的节点个数为$n/k$(向下取整)，那么还有$n\%k$个节点没有处理，为了满足题意，我们需要将这$n\%k$个节点分散到链表的前几部分，即前$n\%k$个链表每个都具有$n/k+1$个元素，剩下的$k-n\%k$个链表每个都具有$n/k$个元素。注意如果最后不够$k$个部分，需要补上空节点。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    vector<ListNode*> splitListToParts(ListNode* root, int k) {
        vector<ListNode*> ans;
        int n=0;
        ListNode *cur=root;
        while(cur&&++n)
            cur=cur->next;
        int num=n/k;
        //分为cnt1个长为num+1的链表 和 cnt2个长为num的链表
        int cnt1=n%k,cnt2=k-cnt1;
        cur=root;
        ListNode *nxt;
        while(cur){
            ans.push_back(cur);
            nxt=cur;
            int tar,val=0;
            if(cnt1)
                --cnt1,tar=num+1;
            else
                --cnt2,tar=num;
            while(++val<tar)
                nxt=nxt->next;
            cur=nxt->next;
            nxt->next=nullptr;
        }
        while(cnt2--)
            ans.push_back(nullptr);
        return ans;
    }
};