HDU 6278

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/liufengwei1/article/details/83017516

上半年湘潭的C题,当时因为太菜了忘记了主席树,于是用莫队二分树状数组水过去了n(sqrt(n)logn+log^2n)的复杂度,现在用主席树就是nlog^2n的复杂度,结果也用了1.1s,当时肯定是因为数据水,没卡莫队到nsqrt(n)所以3s水过去了。

把权值主席树,二分一下答案,然后就记录一蛤从>=mid的地方到nn在L到R区间有多少个数字就行了,主席树裸题。

#include<bits/stdc++.h>
#define maxl 100010

using namespace std;

int n,nn,q,tot,cas;
int a[maxl],b[maxl],rt[maxl];
struct node
{
	int ls,rs,sum;
}tree[maxl<<5];

inline void insert(int &x,int y,int num,int l,int r)
{
	tree[++tot]=tree[y];x=tot;
	tree[x].sum++;
	if(l==r)
		return;
	int mid=(l+r)>>1;
	if(num<=mid)
		insert(tree[x].ls,tree[y].ls,num,l,mid);
	else
		insert(tree[x].rs,tree[y].rs,num,mid+1,r);
}

inline void prework()
{
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]),b[i]=a[i];
		rt[i]=0;
		tree[0].ls=tree[0].rs=tree[0].sum=0;
	}
	sort(b+1,b+1+n);
	nn=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++)
		a[i]=lower_bound(b+1,b+1+nn,a[i])-b;
	tot=0;
	for(int i=1;i<=n;i++)	
		insert(rt[i],rt[i-1],a[i],1,nn);
}

inline int query(int i,int j,int L,int R,int l,int r)
{
	if(l==L && r==R)
		return tree[j].sum-tree[i].sum;
	int mid=(l+r)>>1,sum;
	if(R<=mid)
		sum=query(tree[i].ls,tree[j].ls,L,R,l,mid);
	else
	if(L>mid)
		sum=query(tree[i].rs,tree[j].rs,L,R,mid+1,r);
	else
	{
		sum=query(tree[i].ls,tree[j].ls,L,mid,l,mid);
		sum+=query(tree[i].rs,tree[j].rs,mid+1,R,mid+1,r);
	}
	return sum;
}

inline int find(int L,int R)
{
	int l=1,r=b[nn],mid,sum,num;
 	while(l+1<r)
	{
		mid=(l+r)>>1;
		num=lower_bound(b+1,b+1+nn,mid)-b;
		sum=query(rt[L-1],rt[R],num,nn,1,nn);
		if(sum>=mid)
			l=mid;
		else
			r=mid;
	}
	num=lower_bound(b+1,b+1+nn,r)-b+1;
	sum=query(rt[L-1],rt[R],num,nn,1,nn);
	if(sum>=r)
		return r;
	else
		return r-1;
}

inline void mainwork()
{
	int l,r;
	for(int i=1;i<=q;i++)
	{
		scanf("%d%d",&l,&r);
		printf("%d\n",find(l,r));
	}
}

int main()
{
	while(~scanf("%d%d",&n,&q))
	{
		prework();
		mainwork();
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/liufengwei1/article/details/83017516