【LeetCode：2779】数组的最大美丽值（Java）

dawn1912282024-08-25 20:37

题目链接

2779. 数组的最大美丽值

题目描述

给你一个下标从 0 开始的整数数组 nums 和一个非负整数 k 。

在一步操作中，你可以执行下述指令：

在范围 [0, nums.length - 1] 中选择一个此前没有选过的下标 i 。

将 nums[i] 替换为范围 [nums[i] - k, nums[i] + k] 内的任一整数。

数组的美丽值定义为数组中由相等元素组成的最长子序列的长度。

对数组 nums 执行上述操作任意次后，返回数组可能取得的最大美丽值。

注意：你只能对每个下标执行一次此操作。

数组的子序列定义是：经由原数组删除一些元素（也可能不删除）得到的一个新数组，且在此过程中剩余元素的顺序不发生改变。

示例 1：

输入：nums = [4,6,1,2], k = 2

输出：3

解释：在这个示例中，我们执行下述操作：

选择下标 1 ，将其替换为 4（从范围 [4,8] 中选出），此时 nums = [4,4,1,2] 。
选择下标 3 ，将其替换为 4（从范围 [0,4] 中选出），此时 nums = [4,4,1,4] 。
执行上述操作后，数组的美丽值是 3（子序列由下标 0 、1 、3 对应的元素组成）。
可以证明 3 是我们可以得到的由相等元素组成的最长子序列长度。
示例 2：

输入：nums = [1,1,1,1], k = 10

输出：4

解释：在这个示例中，我们无需执行任何操作。

数组 nums 的美丽值是 4（整个数组）。

提示：

1 <= nums.length <= 10⁵

0 <= nums[i], k <= 10⁵

求解思路

排序 + 滑动窗口：已知序列的顺序对最后的结果并无影响，因此我们可以对数组进行排序。假设最后的美丽值为 x x x，那么经过操作可以变为 x x x 的数值范围为 [ x − k , x + k ] [x-k,x+k] [x−k,x+k]。因此对于排序后的数组，我们只需要找出最大值和最小值差为 2 k 2k 2k 的最长连续子序列即可。
令滑动窗口的右边界为 i i i，左边界为 j j j。从 0 0 0 开始依次枚举直到找到最大值即可。

实现代码

java 复制代码

class Solution {
    public int maximumBeauty(int[] nums, int k) {
        int ans = 0;
        Arrays.sort(nums);
        for (int i = 0, j = 0; i < nums.length; i ++) {
            while (nums[i] - nums[j] > 2 * k) {
                ++ j;
            }
            ans = Math.max(ans, i - j + 1);
        }
        return ans;
    }
}