[LeetCode] 1480. Running Sum of 1d Array

問題

https://leetcode.com/problems/running-sum-of-1d-array/description/

解答と解説

class Solution:
    def runningSum(self, nums: List[int]) -> List[int]:
        res = [nums[0]]
        for i in range(1, len(nums)):
            res.append(res[-1] + nums[i])
        return res

numsの要素数をnとした時、

• Time Complexity: O(n)
• Space Complexity: O(1)

Prefix Sumの典型パターン。

最初の要素をresに入れておくことでres[-1]の値参照時に場合分けが不要になる。