每日一题-303. 区域和检索 - 数组不可变

题目

链接

提交的代码

class NumArray {
public:
    vector<int> data;
    NumArray(vector<int>& nums) {
        data.resize(nums.size());
        for(int i=0;i<nums.size();i++)
            data[i]=nums[i];
    }
    
    int sumRange(int i, int j) {
        int sum=0;
        for(int k=i;k<=j;k++)
            sum+=data[k];
        return sum;
    }
};

其实一开始没想那么多，先直接过了这题easy
发现时间复杂度比较高后考虑优化

优化题解

前缀和
a[i]=presum[i]-presum[i-1]
要注意处理i-1，以及题目中sum.size()可以等于0
即这种情况：
["NumArray"]
[[[]]]

class NumArray {
public:
    vector<int> presum;
    NumArray(vector<int>& nums) {
        presum.resize(nums.size());
        
        if(!nums.empty())
            presum[0]=nums[0];

        for(int i=1;i<nums.size();i++){
            presum[i]=presum[i-1]+nums[i];
        }
    }
    
    int sumRange(int i, int j) {
        if(i==0)
            return presum[j];
        else
            return presum[j]-presum[i-1];
    }
};

官方题解的思路是使得presum的大小是nums.size()+1，方便处理i=0的情况同时也避免了nums为空的情况