leetcode【152周賽】-----5174. 健身計劃評估

一、題目描述

二、分析

        這道題實際上是一道滑動窗口的應用的題，也就是維護一個大小爲K的窗口，計算其中的值，而後再對值進行其餘題目須要的計算。滑動窗口的技巧在一些求子串之類的題中很常見。

三、代碼

class Solution {
public:
    int dietPlanPerformance(vector<int>& calories, int k, int lower, int upper) {
        int res=0;
        int ans=0;
        for(int i=0;i<k;++i){
            ans+=calories[i];
        }
        if(ans>upper) res+=1;
        if(ans<lower) res-=1;
        int len=calories.size();
        for(int j=k;j<len;++j){
            ans=ans-calories[j-k]+calories[j];
            if(ans>upper) res+=1;
            if(ans<lower) res-=1;
        }
        return res;
    }
};
//其實也能夠把第一個循環縮小，這樣就不須要在循環外面計算ans以下
class Solution {
public:
    int dietPlanPerformance(vector<int>& calories, int k, int lower, int upper) {
        int res=0;
        int ans=0;
        for(int i=0;i<k-1;++i){
            ans+=calories[i];
        }
        int len=calories.size();
        for(int j=k-1;j<len;++j){
            ans+=calories[j];
            if(ans>upper) res+=1;
            if(ans<lower) res-=1;
            ans-=calories[j-k+1];
        }
        return res;
    }
};