LeetCode-Day64(C++) 594. 最长和谐子序列

594. 最长和谐子序列

和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 1 。

现在，给你一个整数数组 nums ，请你在所有可能的子序列中找到最长的和谐子序列的长度。

数组的子序列是一个由数组派生出来的序列，它可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。

示例 1：

输入：nums = [1,3,2,2,5,2,3,7]
输出：5
解释：最长的和谐子序列是 [3,2,2,2,3]

示例 2：

输入：nums = [1,2,3,4]
输出：2

示例 3：

输入：nums = [1,1,1,1]
输出：0

方法二：哈希映射
在方法一中，我们枚举了 x 后，遍历数组找出所有的 x 和 x + 1。我们也可以用一个哈希映射（HashMap）来存储每个数出现的次数，这样就能在 O(1)O(1) 的时间内得到 x 和 x + 1 出现的次数。

我们首先遍历一遍数组，得到哈希映射。随后遍历哈希映射，设当前遍历到的键值对为 (x, value)，那么我们就查询 x + 1 在哈希映射中对应的值，就得到了 x 和 x + 1 出现的次数。

方法三：哈希映射 + 单次扫描
在方法二中，我们需要对数组进行一次扫描，再对哈希映射进行一次扫描。事实上，我们也可以设计出只进行一次扫描的算法。

我们扫描一次数组，当扫描到元素 x 时，我们首先将 x 加入哈希映射，随后获取哈希映射中 x - 1, x, x + 1 三者出现的次数 u, v, w，那么 u + v 即为 x - 1, x 组成的和谐子序列的长度，v + w 即为 x, x + 1 组成的和谐子序列的长度。假设数组中最长的和谐子序列的最后一个元素在数组中的位置为 i，那么在扫描到 nums[i] 时，u + v 和 v + w 中一定有一个就是答案。因此这种方法可以找到最长的和谐子序列的长度。

class Solution {
public:
    int findLHS(vector& nums) {
        unordered_map map;
        int ans = 0;
        for (int i : nums) {
            map[i]++;
            if(map.count(i - 1))
                ans = max(ans, map[i] + map[i - 1]);
            if(map.count(i + 1))
                ans = max(ans, map[i] + map[i + 1]);
        }
        return ans;
    }
};