leetcode 303. 区域和检索 - 数组不可变 (python)

题目链接

题目描述：
给定一个整数数组 nums，求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和，包含 i, j 两点。

示例：

给定 nums = [-2, 0, 3, -5, 2, -1]，求和函数为 sumRange()

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3
说明:

你可以假设数组不可变。
会多次调用 sumRange 方法。

解题思路：
由于会多次调用sumRange 方法，所以每次调用都从i到j求和用时比较多，可以用一个数组存储前m个数的和，对于指定的i，j做差即可。

class NumArray(object):

    def __init__(self, nums):
        """
        :type nums: List[int]
        """
        self.nums=nums
        self.list=[]
        sum=0
        for num in nums:
            sum+=num
            self.list.append(sum)

    def sumRange(self, i, j):
        """
        :type i: int
        :type j: int
        :rtype: int
        """
        if i==0:
            return self.list[j]
        else:
            return self.list[j]-self.list[i-1]


# Your NumArray object will be instantiated and called as such:
# obj = NumArray(nums)
# param_1 = obj.sumRange(i,j)