这样一道题目:
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.
我的算法一直超时。
开始的时候每一次都重新计算,这是最最差劲的办法,适合几次的查询。
而在近乎遍历的多次查询中时间复杂度接近O( n3 )。
第二种算法如下,通过建立数组提供查询,算法复杂度为O( n2 ),依然超时。
class NumArray {
public:
NumArray(vector<int> &nums) {
}
int sumRange(int i, int j) {
}
};
// Your NumArray object will be instantiated and called as such:
// NumArray numArray(nums);
// numArray.sumRange(0, 1);
// numArray.sumRange(1, 2);
最后看了大神的算法。时间复杂度O(
n
)。
这个问题,直观上讲是获取任意站点之间路程长度的问题。在实际生活中是票价问题等等,做过类似项目的同学应该会有所思路。就是知道起点到达任意点的长度以后,其中的任何长度都可以通过做差直接求出来。
如此一来,我们只需要建立一个一维数组保存起点到其他点的距离就可以了。用时为o(
public class NumArray {
// 7:54pm - 8:01pm
private int[] nums;
private int[] sums;public NumArray(int[] nums) { this.nums = nums; if (nums == null) { return; } sums = new int[nums.length+1]; sums[0] = 0; for (int i=1; i<=nums.length; i++) { sums[i] = sums[i-1] + nums[i-1]; } }
public int sumRange(int i, int j) {
return sums[j+1] - sums[i];
}
}
总结:自己又写了一遍。
数组左边界要么向左窜位置一个,要么用if进行判断,进行特殊处理。