1571 線段樹區間修改 區間最小值
阿新 • • 發佈:2018-12-12
現在有一個序列 a1, a2, ..., ana1, a2, ..., an ,還有m個查詢 lj, rj (1 ≤ lj ≤ rj ≤ n)lj, rj (1 ≤ lj ≤ rj ≤ n) 。對於每一個查詢,請找出距離最近的兩個元素 axax 和 ay (x ≠ y)ay (x ≠ y) ,並且滿足以下條件:
・ lj ≤ x, y ≤ rj;lj ≤ x, y ≤ rj;
・ ax = ay。ax = ay。
兩個數字的距離是他們下標之差的絕對值 |x − y||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。
Sample Input
樣例輸入1
5 3
1 1 2 3 2
1 5
2 4
3 5
Sample Output
樣例輸出1
1
-1
2
題解:線段樹維護區間最小值,區間查詢即可。固定一個右端點,然後把他能帶來的最小值更新,區間查詢即可。
線段樹維護的是1到它上一次出現位置的最小值。仔細想想這樣維護!!!
#include<algorithm> #include<iostream> #include<cstring> #include<cstdio> #include<string> #include<string> #include<vector> #include<queue> #include<stack> #include<cmath> #include<set> #include<map> using namespace std; #define rep(i,a,n) for (int i=a;i<n;i++) #define per(i,a,n) for (int i=n-1;i>=a;i--) #define lson i<<1,l,m #define rson i<<1|1,m+1,r #define oo cout<<"!!!"<<endl; typedef long long ll; typedef unsigned long long ull; #define ms(s) memset(s, 0, sizeof(s)) const int inf = 1e7; //head const int maxn = 1e6+11111; int minv[maxn<<1],lazy[maxn<<1]; void pushup(int i) { minv[i] = min(minv[i<<1],minv[i<<1|1]); } void pushdown(int i,int l,int r) { if(lazy[i]!=inf) { minv[i<<1] = min(minv[i<<1],lazy[i]); minv[i<<1|1] = min(minv[i<<1|1],lazy[i]); lazy[i<<1] = min(lazy[i<<1],lazy[i]); lazy[i<<1|1] = min(lazy[i<<1|1],lazy[i]); lazy[i] = inf; } } void build(int i,int l,int r) { minv[i] = lazy[i] = inf; if(l==r)return; int m = (l+r)>>1; build(lson); build(rson); } void update(int i,int l,int r,int ql,int qr,int v) { if(ql<=l && qr>=r) { minv[i] = min(minv[i],v); lazy[i] = min(lazy[i],v); return; } pushdown(i,l,r); int m = (l+r)>>1; if(ql<=m)update(lson,ql,qr,v); if(qr>m)update(rson,ql,qr,v); pushup(i); } int query(int i,int l,int r,int ql,int qr) { if(ql<=l && qr>=r) return minv[i]; pushdown(i,l,r); int ans = inf; int m = (l+r)>>1; if(ql<=m)ans = min(ans,query(lson,ql,qr)); if(qr>m)ans = min(ans,query(rson,ql,qr)); pushup(i); return ans; } struct node { int v,id; bool operator < (const node& o)const { return v < o.v; } }a[maxn]; struct node1 { int l,r,id; bool operator < (const node1& o)const { return r < o.r; } }ans[maxn]; int res[maxn]; int pre[maxn],b[maxn],d[maxn]; 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 void write(int x){ if (x<0) putchar('-'),x=-x; if (x>=10) write(x/10); putchar('0'+x%10); } int main() { int n,m; cin>>n>>m; rep(i,1,n+1) { a[i].v = read(); a[i].id = i; } rep(i,1,m+1) { ans[i].l = read(); ans[i].r = read(); ans[i].id = i; } sort(a+1,a+1+n); sort(ans+1,ans+1+m); int cnt = 1; b[a[1].id] = 1; rep(i,2,n+1) { if(a[i].v == a[i-1].v)b[a[i].id] = b[a[i-1].id]; else b[a[i].id] = ++cnt; } memset(d,-1,sizeof d); rep(i,1,n+1) { if(d[b[i]]!=-1) { pre[i] = d[b[i]]; d[b[i]] = i; } else d[b[i]] = i; } build(1,1,n); for(int i=1,j=1;i<=n,j<=m;i++) { if(pre[i])update(1,1,n,1,pre[i],i-pre[i]); while(j<=m && i == ans[j].r) { res[ans[j].id] = query(1,1,n,ans[j].l,ans[j].r);//固定右端點 j++; } } rep(i,1,m+1) { if(res[i] == inf) puts("-1"); else write(res[i]),puts(""); } return 0; }