【Leetcode每日笔记】303. 区域和检索 - 数组不可变(Python)
程序员文章站
2022-03-27 10:09:33
文章目录题目解题思路直接法动态规划代码题目给定一个整数数组 nums,求出数组从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i、j 两点。实现 NumArray 类:NumArray(int[] nums) 使用数组 nums 初始化对象int sumRange(int i, int j) 返回数组 nums 从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i、j 两点(也就是 sum(nums[i], nums[i + 1], ... , nums[j]))示例:输入...
题目
给定一个整数数组 nums,求出数组从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i、j 两点。
实现 NumArray 类:
NumArray(int[] nums) 使用数组 nums 初始化对象 int sumRange(int i, int j) 返回数组 nums 从索引 i 到 j(i ≤ j)范围内元素的总和,包含 i、j 两点(也就是 sum(nums[i], nums[i + 1], ... , nums[j]))
示例:
输入: [“NumArray”, “sumRange”, “sumRange”, “sumRange”] [[[-2, 0, 3, -5,
2, -1]], [0, 2], [2, 5], [0, 5]] 输出: [null, 1, -1, -3]解释: NumArray numArray = new NumArray([-2, 0, 3, -5, 2, -1]);
numArray.sumRange(0, 2); // return 1 ((-2) + 0 + 3)
numArray.sumRange(2, 5); // return -1 (3 + (-5) + 2 + (-1))
numArray.sumRange(0, 5); // return -3 ((-2) + 0 + 3 + (-5) + 2 + (-1))提示:
0 <= nums.length <= 104 -105 <= nums[i] <= 105 0 <= i <= j < nums.length 最多调用 104 次 sumRange 方法
解题思路
直接法
直接初始化,;利用sum进行求和
动态规划
初始化时使用列表dp[i]保存前i位的和,求和时可以直接利用dp列表进行运算
代码
class NumArray:
def __init__(self, nums: List[int]):
self.nums = nums
def sumRange(self, i: int, j: int) -> int:
return sum(self.nums[i:j+1])
class NumArray:
def __init__(self, nums: List[int]):
if(not nums):
return
n=len(nums)
self.dp=[0]*(n+1)
self.dp[1]=nums[0]
for i in range(2,n+1):
self.dp[i]=nums[i-1]+self.dp[i-1]
def sumRange(self, i: int, j: int) -> int:
return self.dp[j+1]-self.dp[i]
本文地址:https://blog.csdn.net/qq_36477513/article/details/111943935