Description
Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.
Example:
Given nums = [-2, 0, 3, -5, 2, -1] sumRange(0, 2) -> 1 sumRange(2, 5) -> -1 sumRange(0, 5) -> -3
Note:
You may assume that the array does not change.
There are many calls to sumRange function.
描述
给定一个整数数组 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 方法。
思路
- 使用动态规划.
- 创建一个新数组sum，数组位置i的值表示给定数组0到i的和.这样当要求i到j的值是返回sum[j]-sum[i]即可.
# -*- coding: utf-8 -*- # @Author: 何睿 # @Create Date: 2019-02-10 21:56:48 # @Last Modified by: 何睿 # @Last Modified time: 2019-02-10 22:02:19 class NumArray: def __init__(self, nums: 'List[int]'): # 对给定的数组的前面的数求和 if not nums: self.sum = [] else: self.sum = [nums[0]] * len(nums) for i in range(1, len(nums)): self.sum[i] = self.sum[i - 1] + nums[i] def sumRange(self, i: 'int', j: 'int') -> 'int': # 做差返回对应序列的值 return self.sum[j] - self.sum[i - 1] if i > 0 else self.sum[j]
源代码文件在这里.
