【leetcode每日一题】2022.03.06第2100题适合打劫银行的日子--双指针，正反扫描取交集

lxhseph

于 2022-03-06 19:10:45 发布

阅读量171

点赞数

文章标签： leetcode 算法双指针

本文链接：https://blog.csdn.net/lxhseph/article/details/123315226

版权

取交集的时候用的是set&set不然速度真的过不了有一个10000长度的测试用例

class Solution(object):
    def goodDaysToRobBank(self, security, time):
        """
        :type security: List[int]
        :type time: int
        :rtype: List[int]
        单调栈啊
        双指针吧，递减的话就右指针动，碰到合适的，加入答案
        递增左指针等于右指针
        """
        len_s=len(security)
        def Search_sec(step,l,r):
            set_a=set()
            while 0<=r<len_s:
                if security[r]<=security[r-step]:
                    if (r-l)*step>=time:
                        set_a.add(r)
                else:
                    l=r
                r=r+step
            return set_a
        if time==0:return range(len_s)
        return list(Search_sec(-1,len_s-1,len_s-2)&Search_sec(1,0,1))