输入一个按升序排序的整数数组(可能包含重复数字),你需要将它们分割成几个子序列,其中每个子序列至少包含三个连续整数。返回你是否能做出这样的分割?

 

示例 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]

思路:我们把 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;
    }
}