题目链接
- 一个List对象可以 ‘.bisect(val)’ 获得大于等于val的最小元素
- 这道题目思想值得看一下
from sortedcontainers import SortedList
class Solution:def avoidFlood(self, rains: List[int]) -> List[int]:ans = [1] * len(rains)l = SortedList() # 可以清空池子的天mp = {} # 当前哪一个池子是满的for i, rain in enumerate(rains):if rain == 0:l.add(i) # 没有下雨，则这一天可以用来清空else:ans[i] = -1if rain in mp: # 如果rain这一个池子是满的，则需要在第mp[rain]天之后的一天清空it = l.bisect(mp[rain])if it == len(l):return []empty_day = l[it]ans[empty_day] = rainl.discard(empty_day)mp[rain] = ireturn ans
, 盲注)
