[BZOJ4627][BeiJing2016]回转寿司(线段树)

从左到右处理，设到当前数R的前缀和为cnt[i]，则以i为右端点的合法的区间左端点j必然是L<=cnt[i]-cnt[j-1]<=R，即cnt[i]-R<=cnt[j-1]<=cnt[i]-L。

问题相当于每次单点在cnt[i]处计数器加1，然后区间询问[cnt[i]-R,cnt[i]-L]中的数之和，动态开点线段树即可。

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #define lson ls[x],L,mid
 #define rson rs[x],mid+1,R
 #define rep(i,l,r) for (int i=(l); i<=(r); i++)
 typedef long long ll;
 using namespace std;

 const int N=;
 const ll inf=1e12;
 ll cnt,res,ans,n,rt,lp,rp,nd,x,sm[N],ls[N],rs[N];

 void mdf(ll &x,ll L,ll R,ll pos){
     if (!x) x=++nd;
     if (L==R){ sm[x]++; return; }
     ll mid=(L+R)>>;
     if (pos<=mid) mdf(lson,pos); else mdf(rson,pos);
     sm[x]=sm[ls[x]]+sm[rs[x]];
 }

 ll que(ll x,ll L,ll R,ll l,ll r){
     if (!x) return ;
     if (L==l && r==R) return sm[x];
     ll mid=(L+R)>>;
     if (r<=mid) return que(lson,l,r);
     else if (l>mid) return que(rson,l,r);
         else return que(lson,l,mid)+que(rson,mid+,r);
 }

 int main(){
     freopen("bzoj4627.in","r",stdin);
     freopen("bzoj4627.out","w",stdout);
     scanf("%lld%lld%lld",&n,&lp,&rp); mdf(rt,-inf,inf,);
     rep(i,,n){
         scanf("%lld",&x); cnt+=x; ll res=que(rt,-inf,inf,cnt-rp,cnt-lp);
         ans+=res; mdf(rt,-inf,inf,cnt);
     }
     printf("%lld\n",ans);
     return ;
 }