【Leetcode】594. Longest Harmonious Subsequence

题目地址：

https://leetcode.com/problems/longest-harmonious-subsequence/

给定一个长$n$数组$A$，求其最长的子序列，使得该子序列的振幅恰好是$1$。返回其长度。

先用哈希表$m$存一下每个数出现了多少次，然后遍历哈希表的时候枚举子序列里较小的数$x$，即只要发现$x+1$也在哈希表里，就更新答案为$m[x]+m[x+1]$。代码如下：

import java.util.HashMap;
import java.util.Map;

public class Solution {
    public int findLHS(int[] nums) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
    
        int res = 0;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            int key = entry.getKey(), val = entry.getValue();
            if (map.containsKey(key + 1)) {
                res = Math.max(res, val + map.get(key + 1));
            }
        }
        
        return res;
    }
}

时空复杂度$O(n)$。