题目链接
题意求给区间[L,R]中有少对(i,j)满足i<j且abs(a[i]-a[j])<=k.
首先来说暴力的方法就是离散化,然后用树状数组来维护,但是m次询问,m很大,所以说一定会t。
这里用到了莫对算法分块来降低复杂度,莫队算法的核心思想是对所有查询进行合理的排序,并将本次查询的结果不断进行修改以得到下次查询的结果
AC代码:
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
#define pii make_pair
#define pr pair<int,int>
const int dir[4][2]={0,1,0,-1,1,0,-1,0};
typedef long long ll;
const ll inFF=9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=3e4+5;
int lt[maxn],mt[maxn],rt[maxn];
int a[maxn],b[maxn],c[maxn],be[maxn],len;
int n,m,k,ans,u;
int res[maxn];
struct node
{
int l,r,id;
}s[maxn];
bool cmp(node s,node e)
{
if(be[s.l]==be[e.l]) return s.r<e.r;
return s.l<e.l;
}
void update(int x,int val)
{
while(x<=len)
{
c[x]+=val;
x+=lowbit(x);
}
}
int getsum(int x)
{
int p=0;
while(x>0)
{
p+=c[x];
x-=lowbit(x);
}
return p;
}
void add(int x)
{
ans+=(getsum(rt[x])-getsum(lt[x]));
update(mt[x],1);
}
void del(int x)
{
update(mt[x],-1);
ans-=(getsum(rt[x])-getsum(lt[x]));
}
int main()
{
int l,r;
memset(c,0,sizeof(c));
scanf("%d %d %d",&n,&m,&k);
u=sqrt(n);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
b[i]=a[i];
be[i]=i/u+1;//分块
}
sort(b+1,b+1+n);
len=unique(b+1,b+1+n)-b-1;
for(int i=1;i<=n;i++)
{
lt[i]=lower_bound(b+1,b+1+len,a[i]-k)-b-1;//a[i]-k对应离散后的下标-1
rt[i]=upper_bound(b+1,b+1+len,a[i]+k)-b-1;//a[i]+k对应离散后的下标
mt[i]=lower_bound(b+1,b+1+len,a[i])-b;//a[i]对应的下标
}
for(int i=1;i<=m;i++)
{
scanf("%d %d",&l,&r);
s[i].l=l,s[i].r=r,s[i].id=i;
}
sort(s+1,s+1+m,cmp);//分块排序降低复杂度
l=1,r=0,ans=0;
for(int i=1;i<=m;i++)//开始对每一个区间进行处理,移动左右端点
{
while(l<s[i].l) del(l++);
while(l>s[i].l) add(--l);
while(r>s[i].r) del(r--);
while(r<s[i].r) add(++r);
res[s[i].id]=ans;
}
for(int i=1;i<=m;i++)
printf("%d\n",res[i]);
return 0;
}
莫队算法思想:https://blog.csdn.net/ThinFatty/article/details/72581276