Codeforces-1000F One Occurrence

題目大意:多次查詢一個最長可達1e5的數列中指定區間內只出現一次的數。

這是一道線段樹題。首先想到的是對於每個區間維護其中每個元素的下一個元素出現的位置中最大的一個。nxt維護當前位置的下一個數字相同的位置。每次query返回該區間的線段樹值即可。因爲要求的是數字,所以線段樹不僅要維護當前區間最大的nxt,還要維護nxt最大的這個位置。(ps程序中的nxt和這裏的描述有些出入,nxt[i]維護的是從右往左遍歷是當前值爲i的最靠左的位置)

但是樣例的第二次查詢就可以證僞這種做法。當出現1 1時,我們會選擇後面的位置,而忽略掉前面的1存在的影響。我們需要通過一些操作在一定情況下對形如上述例子中的第2個1的nxt值進行改動。因爲我們維護的是最大值,顯然賦0即可。

那麼什麼情況下需要改呢?假設查詢區間的左端點是固定的,那麼每次我們從右往左遍歷,當該位置的數不是第一次出現時,我們就應該將該位置線段樹中維護的nxt值改成0 。因此,這道題需要離線處理,將所有操作按左端點從大到小排序,每次接着上次查詢的修改從右往左將線段樹更新,對於右邊有相同數字的位置更新其nxt的線段樹值,直到左端點爲止。(講不太清楚,詳見代碼)

#include<bits/stdc++.h>
using namespace std;

#define pb push_back
#define fi first
#define se second
#define ll long long
#define pq priority_queue
#define mp make_pair
#define pii pair<int,int>
#define mod 998244353
#define debug(x) cerr<<#x<<"="<<x<<'\n'
#define lc v*2
#define rc v*2+1

const int maxn=5e5+10;

int n;
int a[maxn],nxt[maxn],cur[maxn];
pair<pii,int> q[maxn];
pii tree[maxn*4];
int ans[maxn];

bool cmp(pair<pii,int> a,pair<pii,int> b) {
	if (a.fi.fi==b.fi.fi) return a.fi.se>b.fi.se;
	return a.fi.fi>b.fi.fi;
}

void update(int v,int l,int r,int pos,int val) {
	if (l==r) {
		tree[v]=mp(val,pos);
		return;
	}
	int mid=l+r>>1;
	if (pos<=mid) update(v*2,l,mid,pos,val);
	else update(v*2+1,mid+1,r,pos,val);
	tree[v]=max(tree[v*2],tree[v*2+1]);
	return;
}

pii query(int v,int cl,int cr,int l,int r) {
	if (cl>=l&&cr<=r) return tree[v];
	int mid=cl+cr>>1;
	pii res=mp(0,0);
	if (mid>=l) res=max(res,query(v*2,cl,mid,l,r));
	if (mid<r) res=max(res,query(v*2+1,mid+1,cr,l,r));
	return res;
}

int main(){
	scanf("%d",&n);
	for (int i=0;i<n;i++) scanf("%d",&a[i]);
	memset(nxt,-1,sizeof(nxt));
	int m;scanf("%d",&m);
	for (int i=0;i<m;i++) {
		int l,r;
		scanf("%d%d",&l,&r);
		q[i].fi.fi=l-1,q[i].fi.se=r-1;
		q[i].se=i;
	}
	sort(q,q+m,cmp);
	int cur=n-1;
	for (int i=0;i<maxn;i++) nxt[i]=maxn;
	for (int i=0;i<m;i++) {
		int l=q[i].fi.fi,r=q[i].fi.se,id=q[i].se;
		for (int j=cur;j>=l;j--) {
			if (nxt[a[j]]!=maxn) update(1,0,n-1,nxt[a[j]],0);
			update(1,0,n-1,j,nxt[a[j]]);
			nxt[a[j]]=j;
		}
		pii tmp=query(1,0,n-1,l,r);
		if (tmp.fi<=r) ans[id]=0;
		else ans[id]=a[tmp.se];
		cur=l-1;
	}
	for (int i=0;i<m;i++) printf("%d\n",ans[i]);
	return 0;
}

 

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