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

题目描述

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

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

问题分析

此题用贪心算法。创建两个哈希map，freq用来建立数和其出现次数的映射，need用来建立可以加在某个连续子序列后的数及其可以出现的次数的映射。我们先将数组中的数统计一遍，保存到freq中。然后遍历数组，对于每个遍历到的数，先判断它出现的次数，如果为0那么继续遍历下一个数；如果need中存在这个数的非0映射，那么说明当前这个数可以加到某个连续子序列的末尾，我们将其加上去，所以将当前数在freq中的映射值和在need中的映射值减1，然后将下一个需要的连续数（比当前数大1）在need中的映射值加1；如果当前数不能连接到其他子序列后面，那么我们看看这个数是否可以成为新的子序列的起点，如果这个数后面两个连续数在freq中的映射值都大于0的话，说明当前数可以作为新的子序列的起点，那么将这三个数在freq中的映射值减1，然后将这三个数下一个需要的连续数在need中的映射值加1；否则以上情况都不满足，直接返回false。如果当遍历数组的for循环退出时，说明该数组可以分割为连续子序列，返回true。

代码实现

class Solution {
public:
    bool isPossible(vector<int>& nums) {
        unordered_map<int, int> freq, need;
        for(int i : nums)
            ++freq[i];
        for(int i : nums){
            if(freq[i] == 0)
                continue;
            else if(need[i]){
                --freq[i];
                --need[i];
                ++need[i+1];
            }
            else if(freq[i+1] && freq[i+2]){
                --freq[i];
                --freq[i+1];
                --freq[i+2];
                ++need[i+3];
            }
            else
                return false;
        }
        return true;
    }
};