1. 程式人生 > >【樹狀陣列】【BJOI2016】回轉壽司

【樹狀陣列】【BJOI2016】回轉壽司


題目告訴我們要求連續區間價值大於等於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);
}