ZOJ-3633-Alice's present

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


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章