https://leetcode-cn.com/problems/split-linked-list-in-parts/
思路:假设链表长度为 n n n,要将其分成 k k k部分,我们可以先使得每部分的节点个数为 n / k n/k n/k(向下取整),那么还有 n % k n\%k n%k个节点没有处理,为了满足题意,我们需要将这 n % k n\%k n%k个节点分散到链表的前几部分,即前 n % k n\%k n%k个链表每个都具有 n / k + 1 n/k+1 n/k+1个元素,剩下的 k − n % k k-n\%k k−n%k个链表每个都具有 n / k n/k n/k个元素。注意如果最后不够 k 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;
}
};