题目描述:
给你一个长度为 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、找连续上升的段。如果段长至少是 k,那么这段中的所有长为 k 的子数组都是符合要求的,子数组的最后一个元素是最大的。
2、具体来说,遍历数组的同时,用一个计数器 cnt 统计连续递增的元素个数:
3、初始化 cnt=0。
4、如果 i=0 或者 nums[i]=nums[i−1]+1,把 cnt 增加 1;否则,把 cnt 置为 1。
5、如果发现 cnt≥k,那么下标从 i−k+1 到 i 的这个子数组的能量值为 nums[i],即 ans[i−k+1]=nums[i]。
代码实现
class Solution {
public:
vector<int> resultsArray(vector<int>& nums, int k) {
std::vector<int> ans(nums.size() - k + 1, -1);
int cnt = 0;
for (size_t i = 0; i < nums.size(); ++i) {
cnt = (i == 0 || nums[i] == nums[i - 1] + 1) ? cnt + 1 : 1;
if (cnt >= k) {
ans[i - k + 1] = nums[i];
}
}
return ans;
}
};
278

被折叠的 条评论
为什么被折叠?



