编程语言
首页 > 编程语言> > 725. 分隔链表难度中等214给你一个头结点为 head 的单链表和一个整数 k ,请你设计一个算法将链表分隔为 k 个连续的部分。每部分的长度应该尽可能的相等:任意两部分的

725. 分隔链表难度中等214给你一个头结点为 head 的单链表和一个整数 k ,请你设计一个算法将链表分隔为 k 个连续的部分。每部分的长度应该尽可能的相等:任意两部分的

作者:互联网

725. 分隔链表

难度中等214

给你一个头结点为 

head

 的单链表和一个整数 

k

 ,请你设计一个算法将链表分隔为 

k

 个连续的部分。

每部分的长度应该尽可能的相等:任意两部分的长度差距不能超过 1 。这可能会导致有些部分为 null 。

这 

k

 个部分应该按照在链表中出现的顺序排列,并且排在前面的部分的长度应该大于或等于排在后面的长度。

返回一个由上述 

k

 部分组成的数组。

 示例 1:

输入:head = [1,2,3], k = 5 输出:[[1],[2],[3],[],[]] 解释: 第一个元素 output[0] 为 output[0].val = 1 ,output[0].next = null 。 最后一个元素 output[4] 为 null ,但它作为 ListNode 的字符串表示是 [] 。 示例 2:

提示:

0 <= Node.val <= 1000

1 <= k <= 50

通过次数41,930提交次数69,669

请问您在哪类招聘中遇到此题?

方法一:拆分链表

题目要求将给定的链表分隔成 kk 个连续的部分。由于分隔成的每个部分的长度和原始链表的长度有关,因此需要首先遍历链表,得到链表的长度 nn。

得到链表的长度 nn 之后,记 \textit{quotient} = \Big\lfloor \dfrac{n}{k} \Big\rfloorquotient=⌊

k

n

⌋,\textit{remainder} = n \bmod kremainder=nmodk,则在分隔成的 kk 个部分中,前 \textit{remainder}remainder 个部分的长度各为 \textit{quotient} + 1quotient+1,其余每个部分的长度各为 \textit{quotient}quotient。

分隔链表时,从链表的头结点开始遍历,记当前结点为 \textit{curr}curr,对于每个部分,进行如下操作:

将 \textit{curr}curr 作为当前部分的头结点;

计算当前部分的长度 \textit{partSize}partSize;

将 \textit{curr}curr 向后移动 \textit{partSize}partSize 步,则 \textit{curr}curr 为当前部分的尾结点;

当 \textit{curr}curr 到达当前部分的尾结点时,需要拆分 \textit{curr}curr 和后面一个结点之间的连接关系,在拆分之前需要存储 \textit{curr}curr 的后一个结点 \textit{next}next;

令 \textit{curr}curr 的 \textit{next}next 指针指向 \text{null}null,完成 \textit{curr}curr 和 \textit{next}next 的拆分;

将 \textit{next}next 赋值给 \textit{curr}curr。

完成上述操作之后,即得到分隔链表后的一个部分。重复上述操作,直到分隔出 kk 个部分,或者链表遍历结束,即 \textit{curr}curr 指向 \text{null}null。

 

``` 

class Solution {
public:
    vector<ListNode*> splitListToParts(ListNode* head, int k) {
        int n = 0;
        ListNode *temp = head;
        while (temp != nullptr) {
            n++;
            temp = temp->next;
        }
        int quotient = n / k, remainder = n % k;

        vector<ListNode*> parts(k,nullptr);
        ListNode *curr = head;
        for (int i = 0; i < k && curr != nullptr; i++) {
            parts[i] = curr;
            int partSize = quotient + (i < remainder ? 1 : 0);
            for (int j = 1; j < partSize; j++) {
                curr = curr->next;
            }
            ListNode *next = curr->next;
            curr->next = nullptr;
            curr = next;
        }
        return parts;
    }
};

标签:curr,next,链表,长度,textit,分隔,部分
来源: https://blog.csdn.net/weixin_41290863/article/details/120422868