【線段樹】【CDQ分治】回轉壽司

題目大意

給你一個序列，求連續子序列和在【L，R】之間的方案數
N≤100000，|Ai|≤100000，0≤L, R≤109.

分析

這是我打的第一題CDQ（太菜了）
我對這題印象很深刻
當時大家有各種做法
好像都是線段樹？
然後這時出現了一股清流
dhr的CDQ分治（orz dhr 好短啊）
然後愉悅的改完後就沒管了

很久以後yzx講CDQ分治時想到我曾經打過一道

先遞歸
再更新答案
然後sort
回溯

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long LL;
int n,l,r,a[100050];
LL s[100050],ans;
void solve(int p,int q){
    if(p>=q)return;
    int mid=(p+q)/2;
    solve(p,mid); solve(mid+1,q);
    int u=p,v=p;
    for(int i=mid+1;i<=q;i++){
        while(u<=mid&&s[i]-s[u]>=l)u++;
        while(v<=mid&&s[i]-s[v]>r)v++;
        ans+=max(u-v,0);
    }
    sort(s+p,s+q+1);
}
int main(){
    scanf("%d%d%d",&n,&l,&r);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        s[i]=a[i]+s[i-1];
    }
    solve(0,n);
    printf("%lld",ans);
}

現在再補上一種值域線段樹的做法
對於前綴和s【i】（i表示以i爲末尾）
滿足

l<=s【j】−s【i】<=r（i<j）

得

s【j】−r<=s【i】<=s【j】−l

將s【i】插入一顆值域線段樹（初始化時插入0）
對於每個s【j】查詢值域在[s【j】-l,s【j】-r]之間的s【i】個數

#include<cstdio>
#define INF 10000000000
#define N 100010
#define mid ((l+r)>>1)
using namespace std;
typedef long long LL;
LL ans,sum,L,R;
int n,u,v,top,p,t[N*40],lc[N*40],rc[N*40],rt;
void Ins(int &x,LL l,LL r){
    if(!x)x=++top;
    t[x]++;
    if(l==r)return;
    if(sum<=mid)Ins(lc[x],l,mid);
    else Ins(rc[x],mid+1,r);
}
int Que(int x,LL l,LL r){
    if(!x)return 0;
    if(L<=l&&r<=R)return t[x];
    int ans=0;
    if(L<=mid)ans+=Que(lc[x],l,mid);
    if(R>mid)ans+=Que(rc[x],mid+1,r);
    return ans;
}
int main(){
    freopen("data.txt","r",stdin);
    scanf("%d%d%d",&n,&u,&v);
    Ins(rt,-INF,INF);
    for(int i=1;i<=n;i++){
        scanf("%d",&p);
        sum+=p;
        L=sum-v,R=sum-u;
        ans+=Que(rt,-INF,INF);
        Ins(rt,-INF,INF);
    }
    printf("%lld\n",ans);
}

還有一種以i爲首的樹狀數組
鏈接
感覺二分可以換成two pointer
或者可以看Orz 曾老師的代碼

#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
inline ll rd()
{
    ll c, r=0, s=1;
    while((c=getchar())<48||57<c) if(c=='-') s=-1;
    while(c>47&&58>c) r=10*r+c-48, c=getchar();
    return s*r;
}

#define mxn 100233
ll n, l, r, i, j;
pair<ll, ll> a[mxn];
#define fir(i) a[i].first
#define sec(i) a[i].second
//第i小的前綴的值與序號 

ll b[mxn];
inline void add(ll i, ll x) {for(;i<n+2; i+=i&-i) b[i]+=x;}
inline ll sum(ll i) {ll r=0; for(;i; i-=i&-i) r+=b[i]; return r;}

ll ans;
inline ll solve(ll m)
{
    if(m<0) return 0; ans=0;
    for(i=1; i<n+2; i++) b[i]=0;
    for(i=j=1; i<n+1; add(sec(++i), -1))
    {
        for(;j<n+1&&fir(j+1)<=fir(i)+m; add(sec(++j), 1));
        ans+=j-i-sum(sec(i));
    }
    return ans;
}

inline void debug()
{
    for(ll ii=0; ii<=fir(n+1); ii++) printf("%I64d %I64d\n", ii, solve(ii));
}

int main()
{
    n=rd(); l=rd(); r=rd(); sec(1)=1;
    for(i=2; i<=n+1; i++) a[i]=make_pair(fir(i-1)+rd(), i);
    sort(a+1, a+n+2);
    printf("%lld\n", solve(r)-solve(l-1));
    //debug();
    return 0;
}