알고리즘/LeetCode
LeetCode 풀이 - 1480. Running Sum of 1d Array
앤테바
2021. 10. 16. 15:51
반응형
Running Sum of 1d Array
문제)
솔루션1)
class Solution:
def runningSum(self, nums: List[int]) -> List[int]:
ret = [nums[0]]
for i in range(1, len(nums)):
ret.append(ret[i-1] + nums[i])
return ret
솔루션2)
from itertools import accumulate
class Solution:
def runningSum(self, nums: List[int]) -> List[int]:
return list(accumulate(nums))
힌트)
itertools.accumulate() 함수는 누적합을 계산
반응형