51nod1571 最近等對

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');
}