hdu6534 Chika and Friendly Pairs(树状数组+莫队+离散化)

题目

n(n<=27000)个数,m(m<=27000)次区间询问

每次给出L,R,如果[L,R]内(含区间端点)有两个下标(i,j)满足i

则称(i,j)数对是友好的,问[L,R]内有多少个数对是友好的

题解

数据范围n,m几万,读入一堆l,r,明显是莫队

考虑i

所以对于每个ai来说,先把三个值都离散化进去,

这样对于aj来讲,就可以在序列中找到[a_{j}-k,a_{j}+k]这一段内树状数组的值了

树状数组维护离散化值域信息,只维护当前询问的[l,r]内的值域信息

因此,在加入点时,在BIT内加入;当删除时,在BIT内删除

删除就是加入的反序啦,考虑统计贡献的过程,统计完贡献再加入

删除这个值,先让这个值删除,再删除其对原来的贡献

心得

听说这个邀请赛卡常,只能用树状数组+莫队,而后才重现了这场比赛

注意,一并处理好每个aj的下界a_{j}-k和上界a_{j}+k

不要等询问时再处理,不然可能会超时,莫队的O(1)转移比较重要,

剩下的就是裸的树状数组的区间查询和单点修改

代码

#include
using namespace std;
typedef long long ll; 
const int maxn=27005*3;
int n,m,k,up;
int b[maxn],cnt;
int l,r;
int pos[maxn],sz;
ll tree[maxn],res;
struct op
{
	int l,v,r;
}a[maxn/3];
struct node
{
	int l,r,id;
	ll ans;
}e[maxn];
bool cmp1(node a,node b)
{
	if(pos[a.l]==pos[b.l])return a.r0;i-=i&-i)
	ans+=tree[i];
	return ans;
}
void add(int pos,int op)
{
	res+=sum(a[pos].r)-sum(a[pos].l-1);
	update(a[pos].v,op);
}
void del(int pos,int op)
{
	update(a[pos].v,-op);
	res-=sum(a[pos].r)-sum(a[pos].l-1);
}
int main()
{
	while(~scanf("%d%d%d",&n,&m,&k))
	{
		res=0;
		sz=(int)sqrt(n); 
		for(int i=1;i<=n;++i)
		pos[i]=1+(i-1)/sz;
		for(int i=1;i<=n;++i)
		{
		 scanf("%d",&a[i].v);
		 b[++cnt]=a[i].v;
		 b[++cnt]=a[i].l=a[i].v-k;
		 b[++cnt]=a[i].r=a[i].v+k;
	    }
	    sort(b+1,b+cnt+1);
	    up=unique(b+1,b+cnt+1)-(b+1);
	    for(int i=1;i<=n;++i)
	    {
	     a[i].l=lower_bound(b+1,b+up+1,a[i].l)-b;
	     a[i].v=lower_bound(b+1,b+up+1,a[i].v)-b;
	     a[i].r=lower_bound(b+1,b+up+1,a[i].r)-b;
	    }
	    for(int i=1;i<=m;++i)
		{
			scanf("%d%d",&e[i].l,&e[i].r);
			e[i].id=i;
		} 
		sort(e+1,e+m+1,cmp1);
		l=1;r=0;
		for(int i=1;i<=m;++i)
		{
			for(;re[i].r;r--)del(r,1);
			for(;le[i].l;l--)add(l-1,1);
			e[i].ans=res;
		}
		sort(e+1,e+m+1,cmp2);
		for(int i=1;i<=m;++i)
		printf("%lld\n",e[i].ans);
	}
	return 0;
}

 

你可能感兴趣的:(树状数组,莫队分块,离散化,莫队分块)