一道让我受益匪浅的题目,第一遍看题目的时候很懵,不知道怎么用代码实现,看了灵神的思路茅塞顿开
给你一个下标从 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 是我们可以得到的由相等元素组成的最长子序列长度。
思想:1.首先题目没有要求输出替换的下标,因此先排序 【1 2 4 6】
2.每一个数都有可能被选中,进行正负k的改变 因此我们画出草图,可以直观地看到每一个数可能变化的范围
3.画一条竖直的线往下切,被切到的个数是多少,说明当k为这个数是,满足条件的数就有几个
4.
排序后,选出的区间是连续的,我们只需考虑最左边的区间 [x−k,x+k] 和最右边的区间 [y−k,y+k],如果这两个区间的交集不为空,那么选出的这些区间的交集不为空。也就是说,要满足
x+k≥y−k
即
y−x≤2k
于是原问题等价于:
排序后,找最长的连续子数组,其最大值减最小值不超过 2k 这里就可以用滑动窗口解决
代码:
class Solution:
def maximumBeauty(self, nums: List[int], k: int) -> int:
nums.sort()
ans=0
left=0
for right,y in enumerate(nums):
while y-nums[left]>2*k:
left+=1
ans= max(ans,right-left+1)
return ans