leetcode327.区间和的个数

1.题目描述

给定一个整数数组 nums，返回区间和在 [lower, upper] 之间的个数，包含 lower 和 upper。
区间和 S(i, j) 表示在 nums 中，位置从 i 到 j 的元素之和，包含 i 和 j (i ≤ j)。

说明:
最直观的算法复杂度是 O(n2) ，请在此基础上优化你的算法。

示例:

输入: nums = [-2,5,-1], lower = -2, upper = 2,
输出: 3 
解释: 3个区间分别是: [0,0], [2,2], [0,2]，它们表示的和分别为: -2, -1, 2。

2.解题思路

计算累积和数组 sums 的，其中 sum[i] = nums[0] + nums[1] + ... + nums[i]，对于某个i来说，只有那些满足 lower <= sum[j] - sum[i] <= upper 的 j 能形成一个区间 [i, j] 满足题意，则有：sum[i] + lower =< sum[j] <= sum[i] + upper，目标就是来找到有多少个这样的 j满足上述条件。从后向前遍历累加和数组，相当于固定sum[i]后，算出有多少的sum[j]满足左右条件。因为必须满足0 =< i <= j，所以sum[j]的范围一定是由sum[i]之后的元素组成的数组。对sum[j]的范围数组排序，l是找数组中第一个大于等于给定值（左条件）的数，而 r 是找数组中最后一个小于等于给定值（右条件）的数，那么两者相减，就是j的个数。

3.代码实现

class Solution:
    def countRangeSum(self, nums: List[int], lower: int, upper: int) -> int:
        p = [0]                             
        for i in nums:
            p += [p[-1] + i]                
        print (p)
        ans = 0
        q = []                              
        for pi in p[:]:                 
            i, j = pi + lower, pi + upper   
            l = bisect.bisect_left(q, i)    
            r = bisect.bisect_right(q, j)  
            ans += r - l                   
            bisect.insort(q, pi)            
        return ans