[leetcode-链表]725. Split Linked List in Parts

Given a (singly) linked list with head node root, write a function to split the linked list into k consecutive linked list "parts".

The length of each part should be as equal as possible: no two parts should have a size differing by more than 1. This may lead to some parts being null.

The parts should be in order of occurrence in the input list, and parts occurring earlier should always have a size greater than or equal parts occurring later.

Return a List of ListNode's representing the linked list parts that are formed.

给定一个(单独)链表与头节点根,写一个函数将链表拆分成k个连续链表“零件”。

每个部件的长度应尽可能相等:没有两个零件的尺寸相差超过1。这可能导致某些部分为空。

在输入列表中,零件应按顺序排列,并且发生的零件应总是具有大于或等于稍后发生的零件的尺寸。

返回ListNode的列表,该列表表示所形成的链表部分。

Examples 1->2->3->4, k = 5 // 5 equal parts [ [1], [2], [3], [4], null ]

Example 1:

扫描二维码关注公众号,回复: 2273608 查看本文章
Input: 
root = [1, 2, 3], k = 5
Output: [[1],[2],[3],[],[]]
Explanation:
The input and each element of the output are ListNodes, not arrays.
For example, the input root has root.val = 1, root.next.val = 2, \root.next.next.val = 3, and root.next.next.next = null.
The first element output[0] has output[0].val = 1, output[0].next = null.
The last element output[4] is null, but it's string representation as a ListNode is [].

输出的输入和每个元素都是ListNoad,而不是数组。

例如,输入根有根。Val= 1,ROOT.NEX.VAL=2,\ROOT.NEX.NEX.VAL=3,ROOT.NEX.NEX.NET= NULL。

第一个元素输出[0 ]具有输出[0 ],Val= 1,输出[0 ],Next=NULL。

最后一个元素输出[4 ]是NULL,但它的字符串表示为ListNoad是[]。

Example 2:

Input: 
root = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10], k = 3
Output: [[1, 2, 3, 4], [5, 6, 7], [8, 9, 10]]
Explanation:
The input has been split into consecutive parts with size difference at most 1, and earlier parts are a larger size than the later parts.

输入已被分割成连续部分,尺寸差异最多为1,并且较早的部分比后面的部分大。

Note:

  • The length of root will be in the range [0, 1000].
  • Each value of a node in the input will be an integer in the range [0, 999].
  • k will be an integer in the range [1, 50].

//================================================================================

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */
/**
 * Return an array of size *returnSize.
 * Note: The returned array must be malloced, assume caller calls free().
 */
struct ListNode** splitListToParts(struct ListNode* root, int k, int* returnSize) {
   
    struct ListNode *pcur = root;
    
    //计算链表长度 
    int len = 0;
    while(pcur)
    { 
        ++len;
        pcur = pcur->next;
    }
       
    //计算每个子链表的长度(长度不够的,分配1个元素NULL) 
    int parts_num = (len < k) ? 1 : (len/k);
    
    //计算前多少个子链表会多分配1个元素 
    int extra = (len < k) ? (0) : (len % k);

    //分配k个子链表的数组(calloc自动清零) 
    struct ListNode** lists = calloc(k , sizeof(struct ListNode*));
    //struct ListNode** lists = (struct ListNode **)malloc(k * sizeof(struct ListNode*));
    //memset(lists, 0, k*sizeof(struct ListNode*));
    *returnSize = k;
    
    //从原本的链表头开始遍历 
    struct ListNode *prev=NULL;
    pcur = root;

    int i, j;
    for (i = 0; i < k; i++, --extra )//k个子链表,前extra个子链表元素个数多1个 
    {
        //把每个子链表头放进链表数组里 
        lists[i] = pcur;
       
        //给每个子链表分配元素 
        for (j = 0 ; j < parts_num + (extra > 0) && pcur ; j++) 
        {
            prev = pcur;      
            pcur = pcur->next;

        }
        
        //每个子链表彼此断开 
        if (prev) 
            prev->next = NULL; //breaking list became a independent list.
    }


    return lists;
}
 

猜你喜欢

转载自blog.csdn.net/qq_20398345/article/details/81096785
今日推荐