LeetCode（303）Range Sum Query - Immutable

题目

Given an integer array nums, find the sum of the elements between indices i and j (i ≤ j), inclusive.

Example:

Given nums = [-2, 0, 3, -5, 2, -1]

sumRange(0, 2) -> 1
sumRange(2, 5) -> -1
sumRange(0, 5) -> -3

Note:
You may assume that the array does not change.
There are many calls to sumRange function.

分析

给定一个整数序列，求指定子序列和。

提示：数组不会发生变化；大量sumRange函数调用。

题目本身非常简单，只需要遍历 $i$ 到 $j$ ，累计得到和即可。但是，这样是TLE的，所给提示也就没有意义了。

所以，题目考察的是效能，换一个方向思考，我们可以存储子序列和，每个下标处的值为$[0,i]$的所有元素和;

那么$[i,j]$子序列和 $= sum[j] - sum[i-1]$；

注意，$i==0$时，直接返回$sum[j]$即可。

AC代码

class NumArray {
public:
    NumArray(vector<int> &nums) {
        if (nums.empty())
            return ;
        else
        {
            sums.push_back(nums[0]);
            //求得给定数列长度
            int len = nums.size();
            for (int i = 1; i < len; ++i)
            {
                sums.push_back(sums[i - 1] + nums[i]);
            }//for
        }
    }

    //计算[i,j]序列和
    int sumRange(int i, int j) {
        if (0 == i)
            return sums[j];
        int len = sums.size();

        if (i < 0 || i >= len || j < 0 || j >= len || i > j)
        {
            return 0;
        }//if
        return sums[j] - sums[i-1];
    }

private:
    //存储数列和
    vector<int> sums;
};

GitHub测试程序源码