1. 程式人生 > >[bzoj3524][Poi2014]Couriers_主席樹

[bzoj3524][Poi2014]Couriers_主席樹

cst zoj build 小結 sort true n+1 i++ tro

Couriers bzoj-3524 Poi-2014

題目大意:給定n個數的序列,詢問區間內是否存在一個在區間內至少出現了(區間長度>>1)次的數。如果有,輸出該數,反之輸出0。

註釋:$1\le n,m\le 5\cdot 10^5$。


想法:主席樹裸題。

主流做法就是弄一個Existence數組詢問有沒有這樣的數,然後查詢區間中位數即可。

但是可以在query的時候強行查詢,因為沒有輸出0,直接輸出即可。

最後,附上醜陋的代碼... ...

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#define N 500010
using namespace std;
int cnt[N<<5],ls[N<<5],rs[N<<5],tot;
int a[N],b[N],root[N];
int build(int l,int r)
{
	int pos=++tot,mid=(l+r)>>1;
	if(l==r) return pos;
	ls[pos]=build(l,mid); rs[pos]=build(mid+1,r);
	return pos;
}
int update(int pre,int l,int r,int k)
{
	int pos=++tot,mid=(l+r)>>1;
	cnt[pos]=cnt[pre]+1;
	ls[pos]=ls[pre]; rs[pos]=rs[pre];
	if(l==r) return pos;
	if(k<=mid) ls[pos]=update(ls[pre],l,mid,k);
	else rs[pos]=update(rs[pre],mid+1,r,k);
	return pos;
}
// bool Existence(int x,int y,int l,int r,int Want)
// {
// 	if(l==r) return (cnt[y]-cnt[x])>=Want;
// 	int mid=(l+r)>>1;
// 	int LR=1;
// 	if(cnt[ls[y]]-cnt[ls[x]]>cnt[rs[y]]-cnt[rs[x]]) LR=0;
// 	else if(cnt[ls[y]]-cnt[ls[x]]<cnt[rs[y]]-cnt[rs[x]]) LR=2;
// 	if(LR==0) return Existence(ls[x],ls[y],l,mid,Want);
// 	if(LR==2) return Existence(rs[x],rs[y],mid+1,r,Want);
// 	return Existence(ls[x],ls[y],l,mid,Want)|Existence(rs[x],rs[y],mid+1,r,Want);
// }
// int query(int x,int y,int l,int r,int k)
// {
// 	int mid=(l+r)>>1;
// 	if(l==r) return b[l];
// 	int dlt=cnt[ls[y]]-cnt[ls[x]];
// 	if(dlt>=k) return query(ls[x],ls[y],l,mid,k);
// 	else return query(rs[x],rs[y],mid+1,r,k-dlt);
// }
int query(int x,int y,int l,int r,int k)
{
	int mid=(l+r)>>1;
	if(l==r) return l;
	int L=cnt[ls[y]]-cnt[ls[x]],R=cnt[rs[y]]-cnt[rs[x]];
	if(L>k) return query(ls[x],ls[y],l,mid,k);
	if(R>k) return query(rs[x],rs[y],mid+1,r,k);
	else return 0;
}
int main()
{
	int n,m; cin >> n >> m ;
	for(int i=1;i<=n;i++) scanf("%d",&a[i])/* ,b[i]=a[i] */;
	sort(b+1,b+n+1); /* int len=unique(b+1,b+n+1)-b-1; */
	root[0]=build(1,n);
	for(int i=1;i<=n;i++)
	{
		// int t=lower_bound(b+1,b+n+1,a[i])-b;
		root[i]=update(root[i-1],1,n,a[i]);
	}
	for(int x,y,i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		// if(Existence(root[x-1],root[y],1,n,(y-x+1)/2+1)) printf("%d\n",query(root[x-1],root[y],1,n,(y-x+1)/2+1));
		// else printf("0\n");
		printf("%d\n",query(root[x-1],root[y],1,n,(y-x+1)>>1));
	}
	return 0;
}

小結:主席樹真強... ...

[bzoj3524][Poi2014]Couriers_主席樹