适合打劫银行的日子

 

---

class Solution:
    def goodDaysToRobBank(self, security: List[int], time: int) -> List[int]:

        security_len=len(security)
        ans_list=[]

        if time==0:
            return [i for i in range(len(security))]

        day=time

        while day+time<security_len:

            
            flag,d=self.FitForRob(security,day,time)

            # 若该天不适合打劫
            if not flag:
                day=d
                continue
            else:
                ans_list.append(day)
                day+=1

                if day==security_len-time:
                    return ans_list

                while security[day]<=security[day-1] and security[day+time]>=security[day+time-1]:
                    ans_list.append(day)
                    day+=1

                    if day==security_len-time:
                        return ans_list

                day+=1
        
        return ans_list

    def FitForRob(self,security,day,time):

        left_pointer=day-time
        right_pointer=day+time

        for d in range(day,right_pointer):
            if security[d]<=security[d+1]:
                continue
            else:
                return False,d+1

        for d in range(left_pointer,day):
            if security[d]>=security[d+1]:
                continue
            else:
                return False,d+1+time
  
        return True,None