比赛的时候没能A出来,现在来补补题
(后来有做了一道类似的题,发现超时了,这道题还要用到树状数组,等学会了再来修改)
代码如下:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
using namespace std;
typedef long long ll;
const int maxn=1e5+5;
int ans[30005],block,temp;
int a[30005],K;
struct query
{
int l;
int r;
int num;
}qy[30005];
bool cmp(query x,query y)
{
if(x.l/block!=y.l/block)
return x.l<y.l;
return x.r<y.r;
}
void widen(int x,int y,int flag) //区间增大,两种类型,left变小,以及right变大
{
if(x<=y)
{
if(x==y) temp=0;
else if(flag==1)
{
for(int i=y;i>x;i--)
if((int)(abs(a[x]-a[i]))<=K)
temp++;
}
else
{
for(int i=x;i<y;i++)
if((int)(abs(a[i]-a[y]))<=K)
temp++;
}
// printf("%d %d %d\n",x,y,temp);
}
}
void lessen(int x,int y,int flag) //区间缩小,两种类型,left变大,以及right变小
{
if(x<=y)
{
if(x==y) temp=0;
else if(flag==1)
{
for(int i=y;i>x;i--)
if((int)(abs(a[x]-a[i]))<=K)
temp--;
}
else
{
for(int i=x;i<y;i++)
if((int)(abs(a[i]-a[y]))<=K)
temp--;
}
//printf("%d %d %d\n",x,y,temp);
}
}
int main()
{
int n,q;
scanf("%d%d%d",&n,&q,&K);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
for(int i=1;i<=q;i++)
{
scanf("%d%d",&qy[i].l,&qy[i].r);
qy[i].num=i;
}
block=sqrt(n);temp=0;
sort(qy+1,qy+1+q,cmp);
int left=1,right=1;
for(int i=1;i<=q;i++)
{
while(left<qy[i].l){ lessen(left++,right,1);
}
while(left>qy[i].l){ widen(--left,right,1);
}
while(right<qy[i].r){ widen(left,++right,2);
}
while(right>qy[i].r){ lessen(left,right--,2);
}
ans[qy[i].num]=temp;
}
for(int i=1;i<=q;i++)
printf("%d\n",ans[i]);
return 0;
}