就记下各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)