推 CanIndulgeMe: 技術大牛 06/21 22:29
推 sustainer123: 大師 救我 06/21 22:39
就加加減減 對ㄚ
度過了一個愉快的禮拜五
def maxSatisfied(self, customers: List[int], grumpy: List[int], minutes: int)
-> int:
sum_cus = sum(customers)
loss_cus = sum([c*g for c,g in zip(customers, grumpy)])
n = len(customers)
keep_prefix = [0 for _ in range(n)]
keep_prefix[0] = customers[0]*grumpy[0]
for i in range(1,n):
keep_prefix[i] = keep_prefix[i-1]+customers[i]*grumpy[i]
max_keep = keep_prefix[minutes-1]
for i in range(minutes, n):
max_keep = max(max_keep, keep_prefix[i]-keep_prefix[i-minutes])
return sum_cus-loss_cus+max_keep
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 125.228.146.144 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1718978785.A.A9A.html