最近等對 51Nod - 1571 線段樹區間修改 區間最小值
阿新 • • 發佈:2018-11-06
現在有一個序列 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;
}