[JSOI2018]列队（主席树）

Posted 2021-02-01 Owen_codeisking

跟上次那道列队不一样，但都是九条可怜。。。（吉老师太强了）

在主席树上统计答案，因为值域只有 (10^6) 甚至不用离散化。。。

(Code Below:)

#include <bits/stdc++.h>
#define int long long
using namespace std;
const int maxn=500000+10;
const int lim=1000000;
const int inf=0x3f3f3f3f;
int n,m,a[maxn],Sum[maxn],ans,T[maxn],L[maxn<<5],R[maxn<<5],sum[maxn<<5],siz[maxn<<5],cnt;

inline int read(){
    register int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
    return (f==1)?x:-x;
}

void update(int &now,int pre,int l,int r,int x){
    now=++cnt;L[now]=L[pre];R[now]=R[pre];
    sum[now]=sum[pre]+x;siz[now]=siz[pre]+1;
    if(l == r) return ;
    int mid=(l+r)>>1;
    if(x <= mid) update(L[now],L[pre],l,mid,x);
    else update(R[now],R[pre],mid+1,r,x);
}

int query(int u,int v,int Le,int Ri,int l,int r){
    if(Le > Ri) return 0; 
    if(r <= Ri) return (Le+Ri)*(Ri-Le+1)/2-(sum[v]-sum[u]);
    if(Le <= l) return (sum[v]-sum[u])-(Le+Ri)*(Ri-Le+1)/2;
    int mid=(l+r)>>1,cnt=siz[L[v]]-siz[L[u]];
    return query(L[u],L[v],Le,Le+cnt-1,l,mid)+query(R[u],R[v],Le+cnt,Ri,mid+1,r);
}

signed main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++){
        a[i]=read();
        update(T[i],T[i-1],1,lim,a[i]);
    }
    int l,r,k;
    while(m--){
        l=read(),r=read(),k=read();
        printf("%lld
",query(T[l-1],T[r],k,k+r-l,1,lim));
    }
    return 0;
}