HDU - 6534 Chika and Friendly Pairs 莫队+树状数组+离散化

题目链接：https://cn.vjudge.net/problem/HDU-6534

题意：查询一个区间内，有多少对（x，y），满足|x-y|<=k

题解：莫队搞一下，要先离散化，预处理出每个数x的离散化后的值，和x-k,x+k离散化后的范围，这样加入或去掉某个数，树状数组单点更新一下，区间查询即可，从这个题改变一下自己的写法，莫队的时候直接一个循环就行了。

#include<bits/stdc++.h>
using namespace std;
#define lowbit(x) x&(-x)
const int N=27010;
int CM;
struct node{
	int l,r;
	int id;
	bool operator <(const node &x)const
	{
		if(l/CM != x.l/CM) return l/CM<x.l/CM;
		else return r<x.r;
	}
}q[N];
int n,m,k;
int sum[N];
int a[N],b[N];
int pos[N],l[N],r[N];
int ans[N];
int len;
void update(int pos,int val)
{
	while(pos<=len)
	{
		sum[pos]+=val;
		pos+=lowbit(pos);
	}
}
int query(int pos)
{
	int res=0;
	while(pos)
	{
		res+=sum[pos];
		pos-=lowbit(pos);
	}
	return res;
}
int main()
{
	int ll,rr;
	int res;
	while(~scanf("%d%d%d",&n,&m,&k))
	{
		CM=(int)sqrt(n*1.0);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			b[i]=a[i];	
			sum[i]=0;
		}
		sort(b+1,b+1+n);
		len=unique(b+1,b+1+n)-(b+1);
		for(int i=1;i<=n;i++)
		{
			pos[i]=lower_bound(b+1,b+1+len,a[i])-b;
			l[i]=lower_bound(b+1,b+1+len,a[i]-k)-b;
			r[i]=upper_bound(b+1,b+1+len,a[i]+k)-b-1;
		}
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d",&q[i].l,&q[i].r);
			q[i].id=i;
		}
		sort(q+1,q+1+m);
		ll=1,rr=0;
		res=0;
		for(int j=1;j<=m;j++)
		{
			while(q[j].l<ll)
			{
				ll--;
				res+=query(r[ll])-query(l[ll]-1);
				update(pos[ll],1); 
			}
			while(q[j].l>ll)
			{
				update(pos[ll],-1);
				res-=query(r[ll])-query(l[ll]-1);
				
				ll++;
			}
			while(q[j].r>rr)
			{
				rr++;
				res+=query(r[rr])-query(l[rr]-1);
				update(pos[rr],1);
			}
			while(q[j].r<rr)
			{
				update(pos[rr],-1);
				res-=query(r[rr])-query(l[rr]-1);
				
				rr--;
			}
			ans[q[j].id]=res;
		}
		for(int i=1;i<=m;i++)
		printf("%d\n",ans[i]);
	}	
	return 0;
}