脑筋急转弯&滑动窗口-leetcode-594. 最长和谐子序列

594. 最长和谐子序列

题目描述

和谐数组是指一个数组里元素的最大值和最小值之间的差别 正好是 1 。
现在，给你一个整数数组 nums ，请你在所有可能的子序列中找到最长的和谐子序列的长度。
数组的子序列是一个由数组派生出来的序列，它可以通过删除一些元素或不删除元素、且不改变其余元素的顺序而得到。

哈希

一开始有点被绕进去了，后来发现就是 求 nums 升序后 相邻元素的个数总和的最大值，即可。

思路1：（哈希）

• 使用 map 统计每个元素出现次数；
• 遍历 map，查看 key + 1 是否在 map 中
  • 若 key + 1在 map 中，则更新 res = Math.max(res, map.get(key) + map.get(key + 1));
  • 否则，继续遍历

class Solution {
    public int findLHS(int[] nums) {
        Map<Integer, Integer> map = new HashMap<>();
        // 统计次数
        for (int i : nums) map.put(i, map.getOrDefault(i, 0) + 1);
        int res = 0;
        for (int key : map.keySet()) {
            if (map.containsKey(key + 1)) { // 判断`key + 1` 是否在 map 中
                res = Math.max(res, map.get(key) + map.get(key + 1));
            }
        }
        return res;
    }
}

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$

排序 + 滑动窗口（双指针）

参考彤哥题解

使用 map 的话 空间浪费过多，可以使用 排序 + 滑动窗口

思路2：

• 先排序；
• 使用滑动窗口维护可能称为 和谐子序列的 窗口：
  • 如果窗口内极差在1之内，则扩大窗口；
  • 否则，则缩小窗口

class Solution {
    public int findLHS(int[] nums) {
        Arrays.sort(nums); // 排序
        int left = 0;
        int right = 0;
        int res = 0;
        while (right < nums.length) {
            while (nums[right] - nums[left] > 1) { // 窗口内极差大于 1，则缩小窗口（窗口左端向右移）
                left++;
            }
            if (nums[right] - nums[left] == 1) { // 更新结果
                res = Math.max(res, right - left + 1);
            }
            right++; // 扩大窗口
        }
        return res;
    }
}

• 时间复杂度：$O(nlogn)$ （快排）
• 空间复杂度：$O(logn)$ （快排 栈空间）