LeetCode 303 Range Sum Query

翻译

给定一个整型数组nums，找出在索引i到j（i小于等于j）之间（包含i和j）的全部元素之和。

比如：
给定nums = [-2,0,3,-5,2,-1]

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

批注：
你能够假定这个数组不会改变。
这里会有非常多次对sumRange函数的调用。

原文

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.

分析

一開始我还以为这个题有多简单呢，就写了以下这个代码：

class NumArray {
public:
    vector<int> numArray;
    NumArray(vector<int> &nums) {
        numArray = nums;
    }
    int sumRange(int i, int j) {
        int sum = 0;
        while (i < j) {
            sum += numArray[i++];
        }
        return sum;
    }
};

由于不知道须要构造函数干嘛，所以硬生生的加了这么一句进去，事实上全然没用。

功能是实现了。可是面对题目变态的測试用例，果然还是崩了。

至于题目測试用例有多变态，数据来说话：一共214513个字符，当中包含了nums的数据，也包含了函数调用，nums的长度没去算，可是sumRange函数的调用次数高达10000次。

看到这个測试用例，就明确要针对函数调用做优化，一个不错的办法是将全部的返回的值：

从0到1的全部元素之和，
从0到2的全部元素之和，
从0到10000的全部元素之和，
………………

然后计算i和j之间全部元素之和的时候，（由于包含了i和j），也就是求从0到j的全部元素之和减去从0到i-1的全部元素之和。

所以代码就例如以下所看到的了。可是时间上还是使用了624ms叻。

class NumArray {
public:
    map<int, int> rangeSum;
    NumArray(vector<int> &nums) {
        int sum = 0;
        for (int i = 0; i < nums.size(); ++i) {
            sum += nums[i];
            rangeSum.insert(pair<int, int>(i, sum));
        }
    }

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

尽管对底层的详细实现不是非常了解。但由于有索引，所以我自觉用vector的话查找其值来跟高速；并且加入值也更加高速，由于仅仅是一个值而不用是键值对。

详细代码例如以下：

代码

class NumArray {
public:
    vector<int> rangeSum;

    NumArray(vector<int> &nums) {
        int sum = 0;
        for (int i = 0; i < nums.size(); ++i) {
            sum += nums[i];
            rangeSum.push_back(sum);
        }
    }

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