leetcode 303. 区域和检索 - 数组不可变
题目:
给定一个整数数组 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 方法。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/range-sum-query-immutable
思路: 如果直接每次调用都循环求,肯定超时。重点是多次调用。所以给一个nums就提前算好,然后调用时直接用就好。预算时间复杂度O(n),查询时间复杂度O(1),空间复杂度O(n)。
代码:
class NumArray(object):
def __init__(self, nums):
"""
:type nums: List[int]
"""
if nums==[]:
return
self.nums=nums
self.sum=[0]*len(self.nums)
self.sum[0]=self.nums[0]
for i in range(1,len(self.nums)):
self.sum[i]=self.sum[i-1]+self.nums[i]
def sumRange(self, i, j):
"""
:type i: int
:type j: int
:rtype: int
"""
return self.sum[j]-self.sum[i]+self.nums[i]
# Your NumArray object will be instantiated and called as such:
# obj = NumArray(nums)
# param_1 = obj.sumRange(i,j)
本博客为原创作品,欢迎指导,转载请说明出处,附上本文链接,谢谢!