适合打劫银行的日子

题目
你和一群强盗准备打劫银行。给你一个下标从 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;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值