ZOJ-3633-Alice's present
http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3633
給定一個序列,每次詢問一個區間內,從右往左看,第一個重複的
線段樹,這題做了好長時間,感謝cxlove牛
用一個數組標記和當前數字相等的最右邊的數字的位置,若沒有則記爲-1,比如1 2 3 1 2 3這段數對應的標記數組爲-1 -1 -1 1 2 3,用線段樹記錄區間的標記數組的最大值,該值即爲從右向左第一個重複的數字的前一次出現位置
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<map>
using namespace std;
#define N 500005
map<int,int>mp;
int a[N];
int b[N];
struct cam
{
int x;
int y;
int Max;
}list[N*4];
int max(int x,int y)
{
return x>y?x:y;
}
void build(int k,int x,int y)
{
list[k].x=x;
list[k].y=y;
if(x==y)
{
list[k].Max=a[x];
return;
}
int mid=(x+y)/2;
build(k<<1,x,mid);
build(k<<1|1,mid+1,y);
list[k].Max=max(list[k<<1].Max,list[k<<1|1].Max);
}
int query(int k,int x,int y)
{
if(list[k].x==x&&list[k].y==y)
return list[k].Max;
int mid=(list[k].x+list[k].y)/2;
if(y<=mid)
return query(k<<1,x,y);
else if(x>mid)
return query(k<<1|1,x,y);
else
return max(query(k<<1,x,mid),query(k<<1|1,mid+1,y));
}
int main()
{
int n,m,i;
int x,y,ans;
while(scanf("%d",&n)!=EOF)
{
mp.clear();
for(i=1;i<=n;i++)
{
int k;
scanf("%d",&b[i]);
k=b[i];
if(mp.find(k)==mp.end())
{
mp.insert(make_pair(k,i));
a[i]=-1;
}
else
{
a[i]=mp[k];
mp[k]=i;
}
}
build(1,1,n);
scanf("%d",&m);
while(m--)
{
scanf("%d%d",&x,&y);
ans=query(1,x,y);
if(ans==-1||ans<x)
printf("OK\n");
else
printf("%d\n",b[ans]);
}
printf("\n");
}
return 0;
}