题目描述
给定一个整数数组 nums,求出数组从索引 i 到 j (i ≤ j) 范围内元素的总和,包括 i, j 两点。
说明:
- 假设数组不可变
- 会多次调用sumRange方法
思路
说明中提到:会多次调用sumRange方法,所以每次都逐个相加计算子区间的和不是理想的做法,可以借助缓存来保存区域和。
若使用二维数组dp,dp[i][j]表示 i 到 j 的区域和,需要 O(n^2)空间复杂度。
考虑以上因素,我们使用一维sum,其中sum[i]表示 0 到 i - 1索引处的累加和,即sum[i] 存储 nums前 i 个元素的和。则易得到 i 到 j 的区域和公式:sum[j + 1] - sums[i]
缓存+动态规划
- 定义sum数组,长度为 n + 1。sum[k] 为 nums[0 … k - 1]的累加和,
- 遍历数组,保存累加和,遍历区间[1, n ]:
sum[i] = sum[i - 1] + nums[i - 1]
- 索引i 到 j 间的区域和公式:
sum[j + 1] - sum[i]
复杂度分析:
时间复杂度: O(n),每次查询只需要O(1).