【樹狀陣列】【BJOI2016】回轉壽司
阿新 • • 發佈:2018-12-30
題目告訴我們要求連續區間價值大於等於l,小於等於r的數量,我們考慮使用字首和維護
以sum[i]為結尾的符合條件的區間數量即sum[i]-r至sum[i]-l這段區間中所包含的前面的字首和的數量
我們考慮使用陣列離散化維護
不把sum[i]-l和sum[i]-r加入也可以,程式碼中為避免麻煩加入了
#include<iostream> #include<iomanip> #include<cstdio> #include<cstring> #include<cmath> #include<algorithm> #include<queue> #include<map> using namespace std; char c; int read() { int res=0,bj=1; while(c<'0'||c>'9'){if(c=='-') bj=-1; c=getchar();}; while(c>='0'&&c<='9'){res=(res<<3)+(res<<1)+(c^48); c=getchar();}; return res*bj; } int n,l,r,N,tree[300005]; long long sum[100005],a[300005],ans; int find(int x) { int l=1,r=N; while(l<r) { int mid=(l+r+1)>>1; if(a[mid]>x) r=mid-1; else l=mid; } return l; } void add(int x,int v) { while(x<=N) tree[x]+=v,x+=(x&-x); } int ask(int x) { int ans=0; while(x) ans+=tree[x],x-=(x&-x); return ans; } int main() { scanf("%d%d%d",&n,&l,&r); a[++N]=0; for(int i=1;i<=n;i++) sum[i]=sum[i-1]+read(),a[++N]=sum[i],a[++N]=sum[i]-l,a[++N]=sum[i]-r-1; sort(a+1,a+N+1); N=unique(a+1,a+N+1)-a-1; for(int i=0;i<=n;i++) { int ls=lower_bound(a+1,a+N+1,sum[i]-r-1)-a; int rs=lower_bound(a+1,a+N+1,sum[i]-l)-a; ans+=ask(rs)-ask(ls); int x=lower_bound(a+1,a+N+1,sum[i])-a; add(x,1); } printf("%lld",ans); }