UOJ 猴子喫香蕉問題

Problem description：

 

Solution

    def maxBananas(self,arr:List[str],K):
        monkeys = []
        bananas = []
        res = 0
        for i in range(len(arr)):
            if monkeys:
                if i-monkeys[0]>K:
                    del monkeys[0]
            if bananas:
                if i-bananas[0]>K:
                    del bananas[0]
            if arr[i]=="M":
                if bananas:
                    del bananas[0]
                    res+=1
                else:
                    monkeys.append(i)
            else:
                if monkeys:
                    del monkeys[0]
                    res+=1
                else:
                    bananas.append(i)
        return res