Description
现在有一个序列 a1, a2, …, an ,还有m个查询 lj, rj (1 ≤ lj ≤ rj ≤ n) 。对于每一个查询,请找出距离最近的两个元素 ax 和 ay (x ≠ y) ,并且满足以下条件:
· lj ≤ x, y ≤ rj;
· ax = ay。
两个数字的距离是他们下标之差的绝对值 |x − y| 。
Input
单组测试数据。
第一行有两个整数n, m (1≤n,m≤5*10^5),表示序列的长度和查询的次数。
第二行有n个整数a1,a2,…,an (-10^9≤ai≤10^9)。
接下来有m行,每一行给出两个整数lj,rj (1≤lj≤rj≤n)表示一个查询。
Output
对于每一个查询,输出最近的距离,如果没有相等的元素,输出-1。
Input示例
样例输入1
5 3
1 1 2 3 2
1 5
2 4
3 5
Output示例
样例输出1
1
-1
2
Solution
这一题也很巧妙
我们先找出一个点的在其左边最靠近的与其相等的点
把操作离线并且按右端点排序
然后从左到右枚举点的编号
每次把一个点的最靠近距离放入线段树
这样子单点修改加区间查询即可
Code
#include <cstdio>
#include <cstring>
#include <algorithm>
#define L rt<<1
#define R rt<<1|1
#define fo(i,a,b) for(i=a;i<b+1;i++)
using namespace std;
const int N=500005,inf=90000000;
int n,m,x,y,i,j,cnt,tr[N<<2];
int left[N],a[N],s[N],last[N],ans[N];
struct edge{
int l,r,id;
} q[N];
int read(){
int sum=0,p=1;
char c=getchar();
while (c<'0'||c>'9') {
if (c=='-') p=-p;
c=getchar(); }
while (c>='0'&&c<='9'){
sum=sum*10+c-'0';
c=getchar();}
return sum*p;
}
inline bool cmp(edge x,edge y){ return x.r<y.r;}
inline void build(int rt,int l,int r){
if (l==r){
tr[rt]=inf; return;
}
int mid=(l+r)>>1;
build(L,l,mid);
build(R,mid+1,r);
tr[rt]=inf;
}
inline void update(int rt,int l,int r,int x,int y){
if (l==r){
tr[rt]=y;
return;
}
int mid=(l+r)>>1;
if (x<=mid) update(L,l,mid,x,y);
else update(R,mid+1,r,x,y);
tr[rt]=min(tr[L],tr[R]);
}
inline int query(int rt,int l,int r,int x,int y){
if (l>=x&&r<=y) return tr[rt];
int mid=(l+r)>>1;
int ans=inf;
if (x<=mid) ans=min(ans,query(L,l,mid,x,y));
if (y>mid) ans=min(ans,query(R,mid+1,r,x,y));
return ans;
}
inline void write(int x){
if (x<0) putchar('-'),x=-x;
if (x>=10) write(x/10);
putchar('0'+x%10);
}
int main(){
//freopen("1.in","r",stdin);
//freopen("1.out","w",stdout);
n=read(),m=read();
fo(i,1,n) a[i]=read(),s[i]=a[i];
fo(i,1,m){
x=read(),y=read();
q[i].l=x,q[i].r=y,q[i].id=i;}
sort(s+1,s+n+1);
sort(q+1,q+m+1,cmp);
cnt=unique(s+1,s+n+1)-s-1;
fo(i,1,n) {
a[i]=lower_bound(s+1,s+cnt+1,a[i])-s;
left[i]=last[a[i]],last[a[i]]=i;}
build(1,1,n);
for(i=1,j=1;i<=n&&j<=m;i++){
if (left[i]) update(1,1,n,left[i],i-left[i]);
while (i==q[j].r&&j<=m){
ans[q[j].id]=query(1,1,n,q[j].l,q[j].r);
j++;}
}
fo(i,1,m) write(ans[i]<inf?ans[i]:-1),putchar('\n');
}