[路飞]_分隔链表

103 阅读2分钟

题目介绍

给你一个头结点为 head 的单链表和一个整数 k ,请你设计一个算法将链表分隔为 k 个连续的部分。

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

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

返回一个由上述 k 部分组成的数组。

示例1

image.png

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

示例2

image.png

输入:head = [1,2,3,4,5,6,7,8,9,10], k = 3
输出:[[1,2,3,4],[5,6,7],[8,9,10]]
解释:
输入被分成了几个连续的部分,并且每部分的长度相差不超过 1 。前面部分的长度大于等于后面部分的长度。

提示:

解题思路

题目要求将链表分成 k 份,每一份的长度尽量相等,即使不相等,任意两部分的长度差距不能超过 1,并且排在前面部分的长度应该大于或等于排在后面的长度

可以先将链表的长度除以 k 份,得到结果的整数部分即为每部分的最小长度,如果余数不为 0,将余数平均分配到前面的几个部分长度中,这样便能达到题目的要求

解题步骤:

  1. 遍历链表,计算链表的长度
  2. 链表长度 除以 要分成 k 份,取整数部分为每一份的最小长度
  3. 计算 链表长度 除以 要分成 k 份 的余数,平均分配到前面的 n 份中,即前面 n 份的长度比后面的长度多 1
  4. 根据每一份的长度进行链表的分隔

解题代码

var splitListToParts = function(head, k) {
    let n = 0, p = head
    const res = []
    // 遍历链表,获取链表长度
    while (p) {
        n++
        p = p.next
    }
    // 获取每一份的最小长度
    var num1 = Math.floor(n / k)
    // 前面 num2 份长度多 1
    var num2 = n % k
    p = head
    for(let i = 0; i < k; i++) {
        res.push(p)
        // 如果 p 已经为空,当前部分的长度必定为 0,不需要继续后面的步骤
        if (!p) continue
        // 前 num2 部分长度+1
        let num3 = i < num2 ? num1 + 1 : num1
        while (--num3) {
            p = p.next
        }
        // 将每一部分的尾节点截断,防止长度超过规定
        const temp = p.next
        p.next = null
        p = temp
    }
    return res
};