力扣练习——5 分隔链表

1.问题描述

给定一个头结点为 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, 它代表了最后一个部分为空链表。

 

示例 2：

输入: 

root = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], k = 3

输出: [[1, 2, 3, 4], [5, 6, 7], [8, 9, 10]]

解释:

输入被分成了几个连续的部分，并且每部分的长度相差不超过1.前面部分的长度大于等于后面部分的长度。

 

说明:

root 的长度范围： [0, 1000].

输入的每个节点的大小范围：[0, 999].

k 的取值范围： [1, 50].

 

可使用以下代码，完成其中的splitListToParts函数，其中形参root指向无头结点单链表，k为分成的结果链表的数目，返回的vector中包含k个头指针，分别指向一个结果链表。

#include<iostream>

#include<vector>

using namespace std;

 

struct ListNode

{

    int val;

    ListNode *next;

    ListNode() : val(0), next(NULL) {}

    ListNode(int x) : val(x), next(NULL) {}

    ListNode(int x, ListNode *next) : val(x), next(next) {}

};

class Solution

{

public:

    vector<ListNode*> splitListToParts(ListNode* root, int k)

    {

             //填充本函数完成功能  

    }

};

ListNode *createByTail()

{

    ListNode *head;

    ListNode *p1,*p2;

    int n=0,num;

    int len;

    cin>>len;

    head=NULL;

    while(n<len && cin>>num)

    {

        p1=new ListNode(num);

        n=n+1;

        if(n==1)

            head=p1;

        else

            p2->next=p1;

        p2=p1;

    }

    return head;

}

void  display(vector<ListNode *> lnVec )

{

    for(int i=0; i<lnVec.size(); i++)

    {

        ListNode *p;

        p=lnVec[i];

        cout<<"head-->";

        while(p!= NULL)

        {

            cout<<p->val<<"-->";

            p=p->next;

        }

        cout<<"tail\n";

    }

}

int main()

{

    vector<int> G;

    int k,data,res;

    ListNode* head = createByTail();

    cin>>k;

    vector<ListNode*> lnVec=Solution().splitListToParts(head,k);

    display(lnVec);

    return 0;

}

2.输入说明

首先输入链表长度len，然后输入len个整数，以空格分隔。

再输入k。

3.输出说明

输出格式见范例，每行输出一个链表的信息。

4.范例

输入

3
1 2 3
5

输出

head-->1-->tail
head-->2-->tail
head-->3-->tail
head-->tail
head-->tail

5.代码

#include<iostream>

#include<vector>

using namespace std;



struct ListNode

{

    int val;

    ListNode *next;

    ListNode() : val(0), next(NULL) {}

    ListNode(int x) : val(x), next(NULL) {}

    ListNode(int x, ListNode *next) : val(x), next(next) {}

};

class Solution

{

public:

    vector<ListNode*> splitListToParts(ListNode* root, int k)

    {
    
        vector<ListNode*>vec_node(k);
        
        //1.计算链表长度
        int len = 0;
        ListNode *p = root;
        while (p)
        {
            len++;
            p = p->next;
        }
        //2.计算每段平均长度和需要增加一个元素的组数
        int length_avg = len / k;//计算每段的平均长度
        int length_add = len % k;//计算需要分到前面各组中的元素个数，即每组都需要增加一个元素
        //3.在指定位置进行链表插入操作
        p = root;
        for (int i = 0; i < k&&p!=NULL; i++) //这里一定要加p!=NULL
        {
             vec_node[i]=p; //链表中的第一个元素
             int tmp_length = length_avg;    
             if (length_add > 0)
             {
                 tmp_length++;
                 length_add -= 1; 
             }
                
             for (int j = 0; j < tmp_length-1; j++) //注意这里必须是tmp_length-1
             { 
                 p = p->next;
             }
             ListNode *next = p->next; //这里next指向新的链表表头，注意这里链表都是没有头节点的！
             p->next = NULL; //注意：这里每个单链表其实要互相断掉，[1,2,3,4]和[5,6,7]和[8,9,10]是不相连的单链表
             p = next;
             //
            // 输入:

             //root = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], k = 3

             //输出: [[1, 2, 3, 4], [5, 6, 7], [8, 9, 10]]
             
        }
        return vec_node;
    }

};

ListNode *createByTail()

{

    ListNode *head;

    ListNode *p1, *p2=NULL;

    int n = 0, num;

    int len;

    cin >> len;

    head = NULL;

    while (n<len && cin >> num)

    {

        p1 = new ListNode(num);

        n = n + 1;

        if (n == 1)

            head = p1;

        else

            p2->next = p1;

        p2 = p1;

    }

    return head;

}

void  display(vector<ListNode *> lnVec)

{

    for (int i = 0; i < lnVec.size(); i++)

    {

        ListNode *p;

        p = lnVec[i];

        cout << "head-->";

        while (p != NULL)

        {

            cout << p->val << "-->";

            p = p->next;

        }

        cout << "tail\n";

    }

}

int main()

{

    vector<int> G;

    int k, data, res;

    ListNode* head = createByTail();

    cin >> k;

    vector<ListNode*> lnVec = Solution().splitListToParts(head, k);

    display(lnVec);

    return 0;

}