Iterate through the array starting from index 1, adding the previous element to the current element at each step. This transforms the array in-place into its running sum.
itertools.accumulate()Leverage Python's built-in itertools.accumulate() to compute the running sum in a clean, functional style. Returns a new list without modifying the original array.
""" 1480.1 - Solution 1 - In-Place Prefix Sum """
#####################################################################################
# Imports
#####################################################################################
from typing import List
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def runningSum(self, nums: List[int]) -> List[int]:
"""Running Sum Function"""
for i in range(1, len(nums)):
nums[i] += nums[i - 1]
return nums
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().runningSum([1, 2, 3, 4]))
print(Solution().runningSum([1, 1, 1, 1, 1]))
print(Solution().runningSum([3, 1, 2, 10, 1]))
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()
""" 1480.2 - Solution 2 - Using itertools.accumulate() """
#####################################################################################
# Imports
#####################################################################################
from typing import List
from itertools import accumulate
#####################################################################################
# Classes
#####################################################################################
class Solution:
"""Solution Class"""
def runningSum(self, nums: List[int]) -> List[int]:
"""Running Sum Function"""
return list(accumulate(nums))
#####################################################################################
# Functions
#####################################################################################
def testcase():
"""Test Function"""
print(Solution().runningSum([1, 2, 3, 4]))
print(Solution().runningSum([1, 1, 1, 1, 1]))
print(Solution().runningSum([3, 1, 2, 10, 1]))
#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
testcase()