和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 1 。
现在,给你一个整数数组 nums ,请你在所有可能的子序列中找到最长的和谐子序列的长度。
数组的子序列是一个由数组派生出来的序列,它可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。
示例 1:
输入:nums = [1,3,2,2,5,2,3,7]
输出:5
解释:最长的和谐子序列是 [3,2,2,2,3]
提示:
1 <= nums.length <= 2 * 10^4
-10^9 <= nums[i] <= 10^9
方法一:TreeMap
观察给定数据范围可知无法利用递归实现,又因题意是查找可能得子序列,即可忽略其他元素(以及子序列的排序),故可以考虑将数组安顺序存入 Map 中,顺序遍历 Map 获取两 Key 对应 Value 和的最大值即为结果,此处我们采用 TreeMap 这一数据结构。
public int findLHS(int[] nums) {
Map<Integer, Integer> map = new TreeMap<>();
for (int i : nums) map.put(i, map.getOrDefault(i, 0)+1);
int max = 0;
Iterator<Map.Entry<Integer, Integer>> iterator = map.entrySet().iterator();
Map.Entry<Integer, Integer> last = iterator.next();
while (iterator.hasNext()) {
Map.Entry<Integer, Integer> next = iterator.next();
if (Math.abs(last.getKey()-next.getKey()) == 1)
max = Math.max(max, last.getValue() + next.getValue());
last = next;
}
return max;
}