给你一个长度为 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]
思路
和昨天的题目是完全一样的,但是数据量变大,双循环暴力过不了了。可以应用昨天的类似双指针的方法,直接上代码。
代码(C++)
class Solution {
public:vector<int> resultsArray(vector<int>& nums, int k) {int n = nums.size();vector<int> ans(n-k+1, -1);int cnt = 0;for (int i = 0; i < n; ++i) {cnt = i == 0 || nums[i] - nums[i-1] != 1 ? 1 : cnt + 1;if (cnt >= k) ans[i-k+1] = nums[i];} return ans;}
};
代码(Python)
class Solution(object):def resultsArray(self, nums, k):""":type nums: List[int]:type k: int:rtype: List[int]"""n = len(nums)res = [-1] * (n - k + 1)cnt = 0for i in range(n):cnt = 1 if i == 0 or nums[i] - nums[i-1] != 1 else cnt +1if (cnt >= k):res[i - k + 1] = nums[i]return res;