给你一个按升序排序的整数数组 num(可能包含重复数字),请你将它们分割成一个或多个子序列,其中每个子序列都由连续整数组成且长度至少为 3 。
如果可以完成上述分割,则返回 true ;否则,返回 false 。
示例 1:
输入: [1,2,3,3,4,5]
输出: True
解释:
你可以分割出这样两个连续子序列:
1, 2, 3
3, 4, 5
示例 2:
输入: [1,2,3,3,4,4,5,5]
输出: True
解释:
你可以分割出这样两个连续子序列:
1, 2, 3, 4, 5
3, 4, 5
示例 3:
输入: [1,2,3,4,4,5]
输出: False
提示:
输入的数组长度范围为 [1, 10000]
通过次数6,785
提交次数14,748
思路:我们把 3 个或更多的连续数字称作 chain。我们从左到右考虑每一个数字 x,如果 x 可以被添加到当前的 chain 中,我们将 x 添加到 chain 中,这一定会比创建一个新的 chain 要更好。为什么呢?如果我们以 x 为起点新创建一个 chain ,这条新创建更短的链是可以接在之前的链上的,这可能会帮助我们避免创建一个从 x 开始的长度为 1 或者 2 的短链。
我们将每个数字的出现次数统计好,记 tails[x] 是恰好在 x 之前结束的链的数目。现在我们逐一考虑每个数字,如果有一个链恰好在 x 之前结束,我们将 x 加入此链中。否则,如果我们可以新建立一条链就新建。我们可以优化额外空间到 O(1),因为我们可以像 方法 1 一样统计数字的出现次数,而且我们只需要知道最后 3 个数字的出现次数即可。
class Solution {
public boolean isPossible(int[] nums) {
Map<Integer,Integer> count=new HashMap<>();
Map<Integer,Integer> tails=new HashMap<>();
for(int i=0;i<nums.length;i++)
count.put(nums[i], count.getOrDefault(nums[i], 0)+1);
for(int x : nums){
if(count.get(x)==0)
continue;
else if(tails.containsKey(x) && tails.get(x)>0) {
tails.put(x, tails.getOrDefault(x, 0)-1);
tails.put(x+1, tails.getOrDefault(x+1, 0)+1);
}
else if(count.containsKey(x+1) && count.get(x+1)>0
&& count.containsKey(x+2) && count.get(x+2)>0) {
count.put(x+1, count.get(x+1)-1);
count.put(x+2, count.get(x+2)-1);
tails.put(x+3, tails.getOrDefault(x+3, 0)+1);
}
else
return false;
count.put(x, count.get(x)-1);
}
return true;
}
}