3254. 长度为 K 的子数组的能量值 I
给你一个长度为 n
的整数数组 nums
和一个正整数 k
。
一个数组的 能量值 定义为:
- 如果 所有 元素都是依次 连续 且 上升 的,那么能量值为 最大 的元素。
- 否则为 -1 。
你需要求出 nums
中所有长度为 k
的
子数组
的能量值。
请你返回一个长度为 n - k + 1
的整数数组 results
,其中 results[i]
是子数组 nums[i..(i + k - 1)]
的能量值。
示例 1:
输入:nums = [1,2,3,4,3,2,5], k = 3
输出:[3,4,-1,-1,-1]
解释:
nums
中总共有 5 个长度为 3 的子数组:
[1, 2, 3]
中最大元素为 3 。[2, 3, 4]
中最大元素为 4 。[3, 4, 3]
中元素 不是 连续的。[4, 3, 2]
中元素 不是 上升的。[3, 2, 5]
中元素 不是 连续的。
示例 2:
输入:nums = [2,2,2,2,2], k = 4
输出:[-1,-1]
示例 3:
输入:nums = [3,2,3,2,3,2], k = 2
输出:[-1,3,-1,3,-1]
提示:
1 <= n == nums.length <= 105
1 <= nums[i] <= 106
1 <= k <= n
思路:
class Solution:def resultsArray(self, nums: List[int], k: int) -> List[int]:ans = [-1]*(len(nums)-k+1) #最早开始于索引0,最晚开始于索引n-kcnt = 0for i,x in enumerate(nums):cnt = cnt + 1 if i == 0 or x == nums[i-1]+1 else 1if cnt >= k:ans[i-k+1] = xreturn ans
在这段代码中,ans 数组被初始化为 -1,这意味着如果某个位置没有找到长度至少为 k 的连续递增子序列,那么该位置将保持 -1。ans 数组的长度是 len(nums) - k + 1,这是因为我们要找的是长度至少为 k 的子序列,所以数组中最早的可能的子序列起始位置是索引 0,而最晚的起始位置是 len(nums) - k(因为从这个位置开始,我们还可以有 k-1 个元素来构成长度为 k 的子序列)。
在代码的循环中,只有当 cnt(连续递增的元素数量)达到或超过 k 时,ans 数组的相应位置才会被更新。如果在整个遍历过程中,cnt 从未达到 k,那么 ans 数组中相应的位置就会保持初始的 -1 值。