bzoj 4627: [BeiJing2016]回转寿司 线段树

题意

给一个序列，求有多少个子串满足其和在[L,R]范围内。
n,|ai|<=100000，L,R<=1e9

分析

刷道水题找找手感。。。

设s[i]表示前缀和。开一棵权值线段树，把整个序列扫一遍，每到一个数就把以这个数结尾对答案的贡献计算一下，然后把s[i]扔进线段树里面就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

const int N=100005;
const LL inf=1e10;

int n,L,R,sz,root;
LL a[N];
struct tree{int s,l,r;}t[N*30];

void ins(int &d,LL l,LL r,LL x)
{
    if (!d) d=++sz;
    t[d].s++;
    if (l==r) return;
    LL mid=(l+r)/2;
    if (x<=mid) ins(t[d].l,l,mid,x);
    else ins(t[d].r,mid+1,r,x);
}

int find(int d,LL l,LL r,LL x,LL y)
{
    if (x>y||!d) return 0;
    if (l==x&&r==y) return t[d].s;
    LL mid=(l+r)/2;
    return find(t[d].l,l,mid,x,min(y,mid))+find(t[d].r,mid+1,r,max(x,mid+1),y);
}

int main()
{
    scanf("%d%d%d",&n,&L,&R);
    for (int i=1;i<=n;i++) scanf("%lld",&a[i]),a[i]+=a[i-1];
    ins(root,0,inf*2,inf);
    LL ans=0;
    for (int i=1;i<=n;i++)
    {
        ans+=find(root,0,inf*2,max((LL)0,a[i]-R+inf),min(inf*2,a[i]-L+inf));
        ins(root,0,inf*2,a[i]+inf);
    }
    printf("%lld",ans);
    return 0;
}