【一天一大 lee】分割数组为连续子序列 (难度:中等) - Day20201204

【一天一大 lee】分割数组为连续子序列 (难度:中等) - Day20201204

题目:

给你一个按升序排序的整数数组 num(可能包含重复数字),请你将它们分割成一个或多个子序列,其中每个子序列都由连续整数组成且长度至少为 3 。

如果可以完成上述分割,则返回 true ;否则,返回 false 。

示例:

  1. 示例 1:
输入: [1,2,3,3,4,5]
输出: True
解释:
你可以分割出这样两个连续子序列 :
1, 2, 3
3, 4, 5
  1. 示例 2:
输入: [1,2,3,3,4,4,5,5]
输出: True
解释:
你可以分割出这样两个连续子序列 :
1, 2, 3, 4, 5
3, 4, 5
  1. 示例 3:
输入: [1,2,3,4,4,5]
输出: False

提示:

  • 输入的数组长度范围为 [1, 10000]

抛砖引玉

思路:

贪心

【一天一大 lee】分割数组为连续子序列 (难度:中等) - Day20201204

从前到后遍历 nums,模拟分割子序列,对应遇到的任意元素,其可以作为新子序列的起点也可以附加到前一个子序列。

以为题目要求子序列最少要三个元素,那么保证子序列满足要求,优先将遇到的子序列附加到前一个子序列:

  1. 遍历 nums 记录每个元素出现的次数

  2. 逐个取 nums 元素 item(map 统计数量减少):

    • item 出现次数和其之后的两个元素数量均大于 0,则他们可以作为一个新子序列存在,新子序列的结尾为 item+3
    • 如果 item 是一个子序列的结尾,那么优先将其附加到上一个子序列,将其后一个元素看做子序列的结尾 item+1
    • 如果 nums 所有元素均可以按照上面逻辑分割成子序列,则返回 true,否则返回 false

注意: 在标记元素作为子序列结尾时,其可以存在也可以不存在,因为前面的序列已经满足题目要求

var isPossible = function(nums) {
    const len = nums.length
    if (len < 3) return false
    let countMap = new Map(),
        map = new Map()
    // 统计每个数字出现的次数
    for (let i = 0; i < len; i++) {
        const num = countMap.get(nums[i]) || 0
        countMap.set(nums[i], num + 1)
    }
    // 模拟分割 优先附加到先前构造的连续序列中
    for (let i = 0; i < len; i++) {
        const item = nums[i],
            // 一组三个连续数组
            num = countMap.get(item) || 0,
            next1 = countMap.get(item + 1) || 0,
            next2 = countMap.get(item + 2) || 0,
            // 动态记录元素作为子序列结束的次数
            end = map.get(item) || 0,
            endNext1 = map.get(item + 1) || 0,
            endNext2 = map.get(item + 3) || 0
        // 如果数字出现次数归零,则进行向后查找
        if (num === 0) continue
        // 遇到可以作为子序列结束的元素,将其附加到前子序列,其下一个元素代替其作为子序列的结尾
        if (end > 0) {
            map.set(item, end - 1)
            map.set(item + 1, endNext1 + 1)
        } else if (next1 > 0 && next2 > 0) {
            // 连续三个数后标记一个可能的子序列结尾
            countMap.set(item + 1, next1 - 1)
            countMap.set(item + 2, next2 - 1)
            map.set(item + 3, endNext2 + 1)
        } else {
            // 如果数组中遇到既不是子序列结尾,右不能组成新子序列的元素,则直接返回false
            return false
        }
        countMap.set(item, num - 1)
    }
    return true
}

博客: 前端小书童

每天的每日一题,写的题解会同步更新到公众号一天一大 lee 栏目
欢迎关注留言

公众号:前端小书童

上一篇:hung-yi lee_p18_图神经网络(cont.)


下一篇:【一天一大 lee】翻转对 (难度:困难) - Day20201128