Leetcode 659. 分割数组为连续子序列 C++

Leetcode 659. 分割数组为连续子序列

题目

给你一个按升序排序的整数数组 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. 输入的数组长度范围为 [1, 10000]

题解

我们先尽可能地构造长的连续子序列。即如果存在nums[i]-1的连续子序列，我们就可能将nums[i]放到这个连续子序列上。否则，我们需要判断是否存在nums[i]+1以及nums[i]+2，存在，则可以构造连续子序列；否则，则return false。
我们用哈希表time记录数字可使用次数，哈希表tail记录每个连续子序列的最后一个数字。详细过程见代码

代码

	bool isPossible(vector<int>& nums) {
        unordered_map<int,int> time;
        unordered_map<int,int> tail;
        int n = nums.size();
        for(int i=0; i<n; i++)
            time[nums[i]]++;
        for(int i=0; i<n; i++){
            if(time[nums[i]] == 0)    continue;		//该数字已全部使用
            if(tail[nums[i]-1]!=0){			//当前数字可以和某个连续子序列进行合并
                time[nums[i]]--;
                tail[nums[i]-1]--;
                tail[nums[i]]++;
            }else{
                if(time[nums[i]+1]!=0 && time[nums[i]+2]!=0){		//可以构造一个新的连续子序列
                    time[nums[i]]--;
                    time[nums[i]+1]--;
                    time[nums[i]+2]--;
                    tail[nums[i]+2]++;
                }else   return false;
            }
        }
        return true;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/split-array-into-consecutive-subsequences
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。