给你一个下标从 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^5
0 <= nums[i], k <= 10^5
第一种思路:
暴力解,直接hashmap开表,key是可能的每个值, val是对应出现频率。
缺点:会超时。
时间复杂度:O(NK)
空间复杂度:O(N)
第二种思路:
扫描线区间法,或者叫上下车记录法,类似题目有:
LeetCode-Python-1094. 拼车
LeetCode-Python-370. 区间加法
特点在于,对于每一个区间,比如这道题里的 [num - k, num + k],我们只需要处理区间头尾,中间的部分不用管,在最后所有区间都处理完了之后再整体扫一遍就可以确定每个点出现的频率。
这道题因为会有负数下标,所以注意整体向右平移 2 * k + 10。
时间复杂度:O(N) + O(max(nums) + K)
空间复杂度:O(max(nums) + K)
class Solution:
def maximumBeauty(self, nums, k):
res = 0
time = [0] * (max(nums) + 2 * k + 10)
for num in nums:
time[num] += 1
time[num + 2 * k + 1] -= 1
for i, x in enumerate(time):
time[i] += time[i - 1]
res = max(res, time[i])
return res