长度为K的子数组的能量值I

2024-11-06

题目:

给你一个长度为 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 <= 500
  • 1 <= nums[i] <= 105
  • 1 <= k <= n

思路:

由于是判断子数组,那么一定是在原数组的基础上通过滑动窗口遍历所有子数组。

由于不是连续且上升的数组,那么能量值就是-1,所以我们可以初始化一个全为-1的答案数组,然后去遍历原数组:
	如果连续上升的元素超过 k 个,那么答案就是子数组的最后一个元素,否则就归为-1

代码:

class Solution {
public:
    vector<int> resultsArray(vector<int>& nums, int k) {
        int n = nums.size();
        vector<int> ans(n - k + 1, -1);
        for (int i = 0, cnt = 1; i < n; i++) {
            if (i > 0 && nums[i] == nums[i - 1] + 1) cnt++;
            else cnt = 1;
            if (cnt >= k) ans[i - k + 1] = nums[i];
        }
        return ans;
    }
};