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

题目

给定一个整数数组 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])）

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/range-sum-query-immutable

方法一、遍历

每次都从i遍历到j计算答案

class NumArray {

private:
    vector<int> nums;    
public:
    NumArray(vector<int>& nums) {
        this->nums = nums;
    }
    
    int sumRange(int i, int j) {
        int res = 0;
        for(int k = i; k <= j; k++)
            res+=nums[k];
        return res;
    }
};

方法二、分块和

将nums以block_size的大小分块计算和，并提前存储到一个数组中

class NumArray {
    private:
    int block_size;
    vector<int> nums;
    vector<int> sums;
    
public:
    NumArray(vector<int>& nums) {
        this->nums = nums;
        block_size = 100;
        int sum = 0;
        int cnt = 0;
        for(int i = 0; i < nums.size(); ++i)
        {
            cnt++;
            sum += nums[i];
            if(cnt==100) 
            {
                cnt = 0;
                sums.push_back(sum);
                sum = 0;
            }
        }
    }
    
    int sumRange(int i, int j) {
        int res = 0;
        int k = i;
        while(k<=j)
        {
             
            // [0,99] [100,199],[200,299]
            if(k%block_size==0 && k+block_size-1 <= j)
            {
                res += sums[k/block_size];
                k += block_size;
            }else
            {
                res += nums[k];
                ++k;
            }
        }
        return res;
    }
};

方法三、前缀和

• 前缀和：前缀和是一个数组的某项下标之前(包括此项元素)的所有数组元素的和。
• 一维情况下： presum[i] = presum[i-1]+num[i]
  本题求的是i-j的和，如果用presum[j]-presum[i] 得到的是i+1到j的和，固用presum[i+1] = presum[i]+nums[i]

class NumArray {
public:
    vector<int> presum;

    NumArray(vector<int>& nums) {
        int n = nums.size();
        presum.resize(n + 1);
        for (int i = 0; i < n; i++) {
            presum[i + 1] = presum[i] + nums[i];
        }
    }

    int sumRange(int i, int j) {
        return presum[j + 1] - presum[i];
    }
};