Leetcode-5174 Diet Plan Performance (Fitness Program Evaluation)

 1 #define _for(i,a,b) for(int i = (a);i < b;i ++)
 2 #define _rep(i,a,b) for(int i = (a);i > b;i --)
 3 
 4 class Solution
 5 {
 6     public:
 7         int dietPlanPerformance(vector<int>& calories, int k, int lower, int upper)
 8         {
 9             long long sum[100003];
10             memset(sum,0,sizeof(sum));
11             sum[1] = calories[0];
12             _for(i,2,calories.size()+1)
13                 sum[i] = sum[i-1]+calories[i-1];
14             
15             int tol = 0;
16             _for(i,0,calories.size()-k+1)
17             {
18                 if(sum[i+k]-sum[i]<lower)
19                     tol --;
20                 else if(sum[i+k]-sum[i]>upper)
21                     ++ Tol ;
22              }
 23              Return Tol;
24          }
 25 };