LeetCode 327. 区间和的个数（离散化+树状数组）

题意：

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

请你以下标 i （0 <= i <= nums.length ）为起点，元素个数逐次递增，计算子数组内的元素和。

当元素和落在范围 [lower, upper] （包含 lower 和 upper）之内时，
记录子数组当前最末元素下标 j ，记作 有效 区间和 S(i, j) 。

求数组中，值位于范围 [lower, upper] （包含 lower 和 upper）之内的 有效 区间和的个数。


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

数据范围:
0 <= nums.length <= 10^4

解法：

lower<=sum[r]-sum[l-1]<=upper

对于固定的r,
sum[l-1]的取值范围为:
sum[r]-upper<=sum[l-1]<=sum[r]-lower.
只需要判断范围内有多少个这样的l即可.

可以用树状数组做,
因为数值可能是负数,所以还要离散化一下.

注意:
数据加减法可能爆int.

code：

#define ll long long
const int maxm=3e5+5;
struct BIT{
    int c[maxm];
    int lowbit(int i){
        return i&-i;
    }
    void add(int i,int t){
        while(i<maxm){
            c[i]+=t,i+=lowbit(i);
        }
    }
    int ask(int i){
        int ans=0;
        while(i){
            ans+=c[i],i-=lowbit(i);
        }
        return ans;
    }
    void init(){
        memset(c,0,sizeof c);
    }
}T;
ll sum[maxm];
ll L[maxm],R[maxm];
ll xx[maxm];
class Solution {
public:
    int countRangeSum(vector<int>& a, int lower, int upper) {
        int n=a.size();
        if(!n)return 0;
        sum[0]=a[0];
        L[0]=sum[0]-upper;
        R[0]=sum[0]-lower;
        for(int i=1;i<n;i++){
            sum[i]=sum[i-1]+a[i];
            L[i]=sum[i]-upper;
            R[i]=sum[i]-lower;
        }
        int num=0;
        xx[++num]=0;
        for(int i=0;i<n;i++){
            xx[++num]=sum[i];
            xx[++num]=L[i];
            xx[++num]=R[i];
        }
        //
        sort(xx+1,xx+1+num);
        num=unique(xx+1,xx+1+num)-xx-1;
        int zero=lower_bound(xx+1,xx+1+num,0)-xx;
        for(int i=0;i<n;i++){
            sum[i]=lower_bound(xx+1,xx+1+num,sum[i])-xx;
            L[i]=lower_bound(xx+1,xx+1+num,L[i])-xx;
            R[i]=lower_bound(xx+1,xx+1+num,R[i])-xx;
        }
        //
        int ans=0;
        T.init();
        T.add(zero,1);
        for(int i=0;i<n;i++){
            ans+=T.ask(R[i])-T.ask(L[i]-1);
            T.add(sum[i],1);
        }
        return ans;
    }
};

---