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

题目

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

每次给出L,R，如果[L,R]内(含区间端点)有两个下标(i,j)满足i<j且|ai-aj|<=K，

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

题解

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

考虑i<j且|ai-aj|<=k，实际上是

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

这样对于aj来讲，就可以在序列中找到这一段内树状数组的值了

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

因此，在加入点时，在BIT内加入；当删除时，在BIT内删除

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

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

心得

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

注意，一并处理好每个aj的下界和上界，

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

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

代码

#include<bits/stdc++.h>
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.r<b.r;
	return a.l<b.l;
}
bool cmp2(node a,node b)
{
	return a.id<b.id;
}
void update(int x,int v)
{
	for(int i=x;i<=up;i+=i&-i)
	tree[i]+=v;
}
ll sum(int x)
{
	ll ans=0;
	for(int i=x;i>0;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(;r<e[i].r;r++)add(r+1,1);
			for(;r>e[i].r;r--)del(r,1);
			for(;l<e[i].l;l++)del(l,1);
			for(;l>e[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;
}