[Leetcode] 594. 最长和谐子序列 java 哈希表

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

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

示例 1:

输入: [1,3,2,2,5,2,3,7]
输出: 5
原因: 最长的和谐数组是：[3,2,2,2,3].

思想：哈希表，求map.get(nums[i])和map.get(nums[i+1])的最大和

class Solution {
    public int findLHS(int[] nums) {
        HashMap<Integer,Integer> map=new HashMap<Integer,Integer>();
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(nums[i])){
                map.put(nums[i],map.get(nums[i])+1);
            }
            else{
                map.put(nums[i],1);
            }
        }
        int max=0;
        for(int num: map.keySet()){
            if(map.containsKey(num+1)){
                int tmp = map.get(num) + map.get(num + 1);
			    max = Math.max(max, tmp);       
            }
                
        }
        return max;        
    }
}

 或者在map.put那部分这样改，更简洁。（别人的）

public int findLHS(int[] nums) {
	Map<Integer, Integer> map = new HashMap<>();
	for (int num : nums) {
		// getOrDefault（JDK 8）
		map.put(num, map.getOrDefault(num, 0) + 1);
	}
	int max = 0;
	for (int num : map.keySet()) {
		// 如果map包括num+1，那么比较当前值
		if (map.containsKey(num + 1)) {
			int tmp = map.get(num) + map.get(num + 1);
			max = Math.max(max, tmp);
		}
	}
	return max;
 
}