[leetcode每日一题] 3255. 长度为 K 的子数组的能量值 II

If_you_love

时间复杂度：$O(N)$；空间复杂度：$O(N)$

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

ssssshi

借用数组再判断不是爆时间就是爆空间，所以改用一个循环

class Solution {
public:
    vector<int> resultsArray(vector<int>& nums, int k) {
        int n=nums.size(),t=1;
        vector<int>ra;
        if(k==1)return nums;
        for(int i=1;i<n;i++)
        {
            if(nums.at(i)-1==nums.at(i-1))t++;
            else t=1;
            if(i>k-2)
            {
                if(t>=k)ra.push_back(nums.at(i));
                else ra.push_back(-1);
            }
        }
        return ra;
    }
};

ygxdsh

vector<int> resultsArray(vector<int>& nums, int k) {
        vector<int> a;
        int s=1;
        if(k==1) return nums;
        for(int i=1;i<nums.size();i++)
        {
            if(nums[i]==nums[i-1]+1) s++;
            else
            {
                s=1;
            }
            if(i>=k-1) 
            {
                if(s>=k) a.push_back(nums[i]);
                else a.push_back(-1);
            }
        }
        return a;
    }

ps:和昨天一样，所以不写思路了（详见昨天）

wszyh

和昨天思路一样不过爆时间了，删了一个循环加了个计数器

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