소프트웨어에 대한 모든 것

LeetCode 풀이 - 1480. Running Sum of 1d Array 본문

알고리즘/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() 함수는 누적합을 계산

반응형
Comments