牛客OI周赛10-提高组：B-Taeyeon的困惑(值域线段树)

做法

单点加单点删，在值域线段树上直接二分就能求值前\(K\)小的和

Code

#include<bits/stdc++.h>
typedef long long LL;
const LL maxn=1e6+9;
inline LL Read(){
    LL x(0),f(1); char c=getchar();
    while(c<'0' || c>'9'){
        if(c=='-') f=-1; c=getchar();
    }
    while(c>='0' && c<='9'){
        x=(x<<3)+(x<<1)+c-'0'; c=getchar();
    }return x*f;
}
LL n,m,K,sum,ans,L,R;
LL a[maxn];
struct D_tree{
    LL nod,root;
    LL size[maxn],son[maxn][2],sum[maxn];
    inline void Update(LL now){
        size[now]=size[son[now][0]]+size[son[now][1]];
        sum[now]=sum[son[now][0]]+sum[son[now][1]];
    }
    void Build(LL &now,LL l,LL r){
        now=++nod;
        if(l==r) return;
        LL mid(l+r>>1);
        Build(son[now][0],l,mid); Build(son[now][1],mid+1,r);
    }
    void Modify(LL now,LL l,LL r,LL x,LL op){
        if(l==r){
            size[now]+=op; 
            sum[now]=size[now]*l; return;
        }
        LL mid(l+r>>1);
        if(x<=mid) Modify(son[now][0],l,mid,x,op);
        else Modify(son[now][1],mid+1,r,x,op);
        Update(now);
    }
    LL Query_K(LL now,LL l,LL r,LL k){
        LL ret(0);
        if(size[now]==k) return sum[now];
        if(l==r){
            return l*k;
        }
        LL mid(l+r>>1);
        if(size[son[now][0]]>=k) return Query_K(son[now][0],l,mid,k);
        else return sum[son[now][0]]+Query_K(son[now][1],mid+1,r,k-size[son[now][0]]);
    }
}T;
int main(){
    n=Read(); m=Read(); K=Read();
    for(LL i=1;i<=n;++i) a[i]=Read();
    T.Build(T.root,L=0,R=100000);
    for(LL i=1;i<=m;++i)
        T.Modify(T.root,L,R,a[i],1);
    for(LL i=1;i<=n-m+1;++i){
        if(i!=1){
            T.Modify(T.root,L,R,a[i-1],-1); T.Modify(T.root,L,R,a[m+i-1],1);
        }
        ans+=T.Query_K(T.root,L,R,K);
    }
    printf("%lld\n",ans);
    return 0;
}