题目
你和一群强盗准备打劫银行。给你一个下标从 0 开始的整数数组 security ,其中 security[i] 是第 i 天执勤警卫的数量。日子从 0 开始编号。同时给你一个整数 time 。
如果第 i 天满足以下所有条件,我们称它为一个适合打劫银行的日子:
第 i 天前和后都分别至少有 time 天。
第 i 天前连续 time 天警卫数目都是非递增的。
第 i 天后连续 time 天警卫数目都是非递减的。
更正式的,第 i 天是一个合适打劫银行的日子当且仅当:security[i - time] >= security[i - time + 1] >= … >= security[i] <= … <= security[i + time - 1] <= security[i + time].
请你返回一个数组,包含 所有 适合打劫银行的日子(下标从 0 开始)。返回的日子可以 任意 顺序排列。
链接:https://leetcode-cn.com/problems/find-good-days-to-rob-the-bank
思路
前缀和
两种方法
第一种前缀和是正规的前缀和
第二种前缀和是中间断开的(一直满足条件,就一直累加,不满足条件,从0开始)
代码
方法一
class Solution {
public:
vector<int> goodDaysToRobBank(vector<int>& security, int time) {
vector<int> ans;
int n=security.size();
if(time==0){
for(int i=0;i<n;i++)
ans.push_back(i);
return ans;
}
if(n<2*time+1)
return {};
vector<int> incre_pre(n,0);
vector<int> decre_pre(n,0);
for(int i=time;i<n-1;i++){
if(security[i]>security[i+1])
incre_pre[i]=1;
incre_pre[i]+=incre_pre[i-1];
}
for(int i=1;i<n-time;i++){
if(security[i-1]<security[i])
decre_pre[i]=1;
decre_pre[i]+=decre_pre[i-1];
}
for(int i=time;i<n-time;i++){
if(decre_pre[i]-decre_pre[i-time] == 0 && incre_pre[i+time-1]-incre_pre[i-1] == 0)
ans.push_back(i);
}
}
};
方法二
class Solution {
public:
vector<int> goodDaysToRobBank(vector<int>& security, int time) {
vector<int> ans;
int n=security.size();
if(time==0){
for(int i=0;i<n;i++)
ans.push_back(i);
return ans;
}
if(n<2*time+1)
return {};
vector<int> incre_pre(n,0);
vector<int> decre_pre(n,0);
for(int i=time;i<n-1;i++){
if(security[i]<=security[i+1])
incre_pre[i]=incre_pre[i-1]+1;
}
for(int i=1;i<n-time;i++){
if(security[i-1]>=security[i])
decre_pre[i]=decre_pre[i-1]+1;
}
for(int i=time;i<n-time;i++){
if(decre_pre[i] >=time && incre_pre[i+time-1] >=time)
ans.push_back(i);
}
return ans;
}
};