LC3254. 长度为 K 的子数组的能量值 I 每日一题记录

3254. 长度为 K 的子数组的能量值 I - 力扣（LeetCode）

给你一个长度为 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] 中元素 不是 连续的。

思路

一开始想法是看到定长字串，定长窗口两次遍历，复杂度为O(n*k)

想了想可以统计进出窗口的元素与上一个的差，记录不合格的元素个数，不合格数量为0记录答案。复杂度为O(n)

class Solution {
public:
    vector<int> resultsArray(vector<int>& nums, int k) {
        vector<int> ans(nums.size()- k +1,0);
        if(k == 1)return nums;
        int n = nums.size();
        int left = 0,sum = 0,error_cnt = 0;
        for(int right = 1;right<n;++right)
        {
            if(nums[right] - nums[right-1]!=1)++error_cnt;
            if(right<k-1)continue;
            //update
            if(error_cnt == 0)ans[left] = nums[right];
            else ans[left] = -1;
            //out
            if(nums[left+1] - nums[left]!=1)--error_cnt;
            ++left;
            
        }
        return ans;
    }
};

看了看灵神的写法 更简一步。遍历nums数组，记录 与上个元素差为1的元素的个数，且当出现不合格元素时清空计数。当计数大于等于k时，记录答案。 这样写同时还省略了出窗口的判断，不如说这个判断本身没什么用。

//作者：灵茶山艾府
//链接：https://leetcode.cn/problems/find-the-power-of-k-size-subarrays-ii/solutions/2884189/on-jian-ji-xie-fa-pythonjavacgo-by-endle-gtek/

class Solution {
public:
    vector<int> resultsArray(vector<int>& nums, int k) {
        vector<int> ans(nums.size() - k + 1, -1);
        int cnt = 0;
        for (int 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;
    }
};

大神不愧是大神，这个想法还能推广开来，在任何需要统计所谓 连续XXX 的条件的字串都可以使用，遇到不合格的元素直接清零重新寻找。