2019湘潭邀请赛 Chika and Friendly Pairs（莫队+树状数组）

解题思路：莫队的复杂度O(mlogn)，然后更新，查询树状数组的复杂度O(logn)，最终复杂度O(mlog^2n)

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 27005
using namespace std;
struct node
{
	int l,r,id;
}q[N];
int be[N],u,vis[N],ans,res[N],bit[N],n,m,k,s[N],s1[N],id,lt[N],rt[N],a[N];
int lowbit(int x)
{
    return x&(-x);
}
int gsum(int i)
{
    int tp=0;
    while(i>0)
    {
        tp+=bit[i];
        i-=lowbit(i);
    }
    return tp;
}
int gadd(int i,int k)
{
    while(i<=N)
    {
        bit[i]+=k;
        i+=lowbit(i);
    }
}
bool cmp(node a,node b)
{
	if(be[a.l]==be[b.l]) return a.r<b.r;
	return a.l<b.l;
}

void add(int pos)
{
    ans+=(gsum(rt[pos])-gsum(lt[pos]));
	gadd(a[pos],1);
}
void del(int pos)
{
	gadd(a[pos],-1);
    ans-=(gsum(rt[pos])-gsum(lt[pos]));
}
int main()
{
	//freopen("t.txt","r",stdin);
	int l,r;
        memset(bit,0,sizeof(bit));
		scanf("%d%d%d",&n,&m,&k);
		u=sqrt(n);
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&s[i]);
			s1[i]=s[i];
			be[i]=i/u+1;
		}
		sort(s1+1,s1+1+n);
		int tp=-1;
		id=1;
		for(int i=1;i<=n;i++)
        {
            if(s1[i]!=tp)
            {
                tp=s1[i];
                s1[id]=s1[i];
                id++;
            }
        }
        /*for(int i=1;i<id;i++)
        printf("%d ",s1[i]);
        printf("\n");*/
        for(int i = 1; i <= n; i ++)
		{
        	lt[i] = (int)(lower_bound(s1 + 1, s1 + id, s[i] - k) - s1)-1;
        	rt[i] = (int)(upper_bound(s1 + 1, s1 + id, s[i] + k) - s1)-1;
        	a[i] = (int)(lower_bound(s1 + 1, s1 + id, s[i]) - s1);
    	}
    	/*for(int i = 1; i <= n; i ++)
    	{
    		printf("%d %d %d %d\n",s[i],lt[i],rt[i],a[i]);
		}*/
		for(int i=1;i<=m;i++)
		{
			scanf("%d%d",&l,&r);
			q[i].l=l;
			q[i].r=r;
			q[i].id=i;
		}
		sort(q+1,q+1+m,cmp);
		l=1,r=0;
		memset(vis,0,sizeof(vis));
		ans=0;
		for(int i=1;i<=m;i++)
		{
			while(l>q[i].l) add(l-1),l--;
			while(r<q[i].r) add(r+1),r++;
			while(l<q[i].l) del(l),l++;
			while(r>q[i].r) del(r),r--;
			res[q[i].id]=ans;
		}
		for(int i=1;i<=m;i++)
		printf("%d\n",res[i]);
	return 0;
}