跳至主要內容

最长和谐子序列

冯泽立大约 2 分钟教学文档哈希表

1.最长和谐子序列

和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 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

解释:

最长和谐子序列是 [1,2][2,3][3,4],长度都为 2。

示例 3:

**输入:**nums = [1,1,1,1]

**输出:**0

不存在和谐子序列。

分析:

我们可以用一个哈希映射来存储每个数出现的次数,这样就能在 O(1) 的时间内得到 x 和 x+1 出现的次数。我们首先遍历一遍数组,得到哈希映射。随后遍历哈希映射,设当前遍历到的键值对为 (x,value),那么我们就查询 x+1 在哈希映射中对应的统计次数,就得到了 x 和 x+1 出现的次数,和谐子序列的长度等于 x 和 x+1 出现的次数之和。

class Solution {
public:
    int findLHS(vector<int>& nums) {
        unordered_map<int, int> cnt;
        int res = 0;
        for (int num : nums) {
            cnt[num]++;
        }
        for (auto [key, val] : cnt) {
            if (cnt.count(key + 1)) {
                res = max(res, val + cnt[key + 1]);
            }
        }
        return res;
    }
};

class Solution {
    public int findLHS(int[] nums) {
        HashMap <Integer, Integer> cnt = new HashMap <>();
        int res = 0;
        for (int num : nums) {
            cnt.put(num, cnt.getOrDefault(num, 0) + 1);
        }
        for (int key : cnt.keySet()) {
            if (cnt.containsKey(key + 1)) {
                res = Math.max(res, cnt.get(key) + cnt.get(key + 1));
            }
        }
        return res;
    }
}

上次编辑于:
贡献者: zilizhou,黄昱皓