题目描述
和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 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
思路
本题实质上是找一个元素及其相邻的x + 1的次数加起来最多的子序列。
方法一:滑动窗口
- 排序不影响,可以先排序,然后用双指针维护一个左右端点小于等于1的滑动窗口,如果等于1就不断地相更新结果即可。
方法二:哈希表计数
- 直接统计每个元素出现的次数,然后再次扫描看x和x + 1的次数累计,不断更新最大值即可
代码
滑动窗口
class Solution {
public int findLHS(int[] nums) {
Arrays.sort(nums);
int n = nums.length;
int res = 0;
//双指针滑动窗口
for (int i = 0, j = 0; j < n; j ++ ) {
//如果差值大于1,就窗口左端移动
while(i < n && nums[j] - nums[i] > 1) i ++;
//如果差值刚好等于1,就更新最大值
if (nums[j] - nums[i] == 1)
res = Math.max(res, j - i + 1);
}
return res;
}
}
哈希表
class Solution {
public int findLHS(int[] nums) {
Map<Integer, Integer> map = new HashMap<>();
//统计所有元素出现的次数
for (int x: nums) {
map.put(x, map.getOrDefault(x, 0) + 1);
}
int ans = 0;
//统计相差1的
for (int x: nums) {
if (map.containsKey(x - 1)) {
ans = Math.max(ans, map.get(x - 1) + map.get(x));
}
}
return ans;
}
}