区间和的个数

给定一个整数数组nums，返回[lower，upper]范围内的区间和数量。
区间和S(i, j)被定义为索引i和j(i ≤ j)之间（包含i和j）的nums中所有元素的和。

样例

样例 1:

输入：[-2, 5, -1],-2,2
输出：3
解释：三个区间分别是: [0,0], [2,2], [0,2] ，它们对应的区间和分别为: -2, -1, 2。

样例 2:

输入：[0,-3,-3,1,1,2],3,5
输出：2
解释：三个区间分别是: [3, 5], [4, 5] ，它们对应的区间和分别为: 4, 3。

注意事项

暴力解法的时间复杂度为O(n^2)。给出的解法必须好于这个方法。

输入测试数据 (每行一个参数)如何理解测试数据？

 

class Solution {
public:
    /**
     * @param nums: a list of integers
     * @param lower: a integer
     * @param upper: a integer
     * @return: return a integer
     */
    int countRangeSum(vector<int> &nums, int lower, int upper) {
        // write your code here
        vector<long long> sum(nums.size()+1,0);
        int ret = 0;
        for(int i = 1; i <= nums.size(); i++)
        {
            sum[i] = sum[i-1] + nums[i-1];
        }
        map<long long,long long> mymap;
        long long mylow;
        long long myhigh;
        for(int i = 0; i < sum.size(); i++)
        {
            mylow = sum[i] - upper;
            myhigh = sum[i] - lower;

            for(long long i = mylow; i <= myhigh; i++)
            {
                if(mymap.count(i) != 0)
                    ret += mymap[i];
            }
            mymap[sum[i]] += 1;
        }
        return ret;
    }
};