推 Smallsh: 大師 10/10 23:18
就記下各modulo目前的sum跟maximum
最後再max一次
吧
一二三四五
def maximumEnergy(self, energy: List[int], k: int) -> int:
mem, maximum = [0]*k, [-10**10]*k
for i in range(len(energy)-1, -1, -1):
cur_idx = i%k
mem[cur_idx] += energy[i]
maximum[cur_idx] = max(maximum[cur_idx], mem[cur_idx])
return max(maximum)
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 220.132.58.28 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Marginalman/M.1760109506.A.3B1.html