Leetcode-303. 区域和检索 - 数组不可变

给定一个整数数组  nums,求出数组从索引 到 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

说明:

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

解答: 

采用概率分布函数的思想,算出sum[i],那么sum(i,j) = sum[j] - sum[i-1]

class NumArray {
public:
    vector<int> dp;
    NumArray(vector<int> nums) {
        int sum = 0;
        for(int i=0;i<nums.size();i++){
            sum += nums[i];
            dp.push_back(sum);
        }
    }
    int sumRange(int i, int j) {
        if(i>=1) return dp[j]-dp[i-1];
        else return dp[j];
    }
};

猜你喜欢

转载自blog.csdn.net/baidu_39622935/article/details/81479569