LeetCode 327 区间和的个数

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

肯定是想要得到一个NlogN的方法。

这道题可以采用归并，线段树，树状数组等方法，这里给出二分搜索题解。

from typing import *
import bisect


class Solution:
    def countRangeSum(self, nums: List[int], lower: int, upper: int) -> int:
        # 相当巧妙的方法
        # 首先转化为前缀和，结果res每次添加的都是
        # 以当前元素为最后一个元素的集合有多少个
        # 主需要在之前的那么多个前缀和中找到在tem_sum-upper和tem_sum-lower中的就可以
        # 已经处理过的前缀和需要排好序
        pre = [0]  # 前缀和，填一个0
        res, tem_sum = 0, 0
        for val in nums:
            tem_sum += val
            res += bisect.bisect_right(pre, tem_sum - lower) - bisect.bisect_left(pre, tem_sum - upper)
            bisect.insort(pre, tem_sum)
        return res