[LeetCode 725]分割链表

[LeetCode 725]C++分割链表

给定一个头结点为 root 的链表, 编写一个函数以将链表分隔为 k 个连续的部分。

每部分的长度应该尽可能的相等: 任意两部分的长度差距不能超过 1，也就是说可能有些部分为 null。

这k个部分应该按照在链表中出现的顺序进行输出，并且排在前面的部分的长度应该大于或等于后面的长度。

返回一个符合上述规则的链表的列表。

举例： 1->2->3->4, k = 5 // 5 结果 [ [1], [2], [3], [4], null ]

示例 1：

输入:
root = [1, 2, 3], k = 5
输出: [[1],[2],[3],[],[]]
解释:
输入输出各部分都应该是链表，而不是数组。
例如, 输入的结点 root 的 val= 1, root.next.val = 2, \root.next.next.val = 3, 且 root.next.next.next = null。
第一个输出 output[0] 是 output[0].val = 1, output[0].next = null。
最后一个元素 output[4] 为 null, 它代表了最后一个部分为空链表。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/split-linked-list-in-parts

方法：划分各个分割链表长度

class Solution {
public:
    vector<ListNode*> splitListToParts(ListNode* root, int k) {
        ListNode* head=root;
        int root_length=0;
        while(head){//求链表长度
            head=head->next;
            ++root_length;
        }

        vector<int>data_length;//分割链表的长度
        data_length.resize(k);
        for(int i=0;i<k;++i){
            data_length[i]=root_length/k;
        }
        for(int i=0;i<root_length%k;++i){
            data_length[i]+=1;
        }

        vector<ListNode*>data;

        ListNode* newHead=new ListNode(0);
        newHead->next=root;
        ListNode* pTail=newHead;

        int index=0;//data_length位置
        int sub_length=0;//子链表长度
        while(root){
            pTail=root;
            root=root->next;
            ++sub_length;
            if(sub_length==data_length[index]){//长度一致，
                ++index;//data_length移动到下一位置
                sub_length=0;//下一链表长度提前置空

                pTail->next=NULL;//子链表尾指空
                data.push_back(newHead->next);//放入子链表

                newHead->next=root;//连接下一子链表首节点
            }
        }
        //root==NULL
        //此时index=data_length.size()
        //k>index则没装满data,push NULL
        while(index<k){
            data.push_back(root);
            ++index;
        }

    return data;
    }
};