整体二分入门及模板

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

hdu 2665

这已经是继学归并树,主席树以来第三次写这个题了,整体二分就是每次二分一个答案m,根据当前的问题,如果在某个问题的区间内,小于等于m的数超过了这个问题的k,那么显然这个问题的答案小于m,因此把这个问题划分到左边,否则划分到右边,然后继续二分,直到当前左区间等于右区间,那么左区间 l 就是当前存在的问题的解,记录下来即可。

#include<cstdio>
#include<algorithm>
#define low(x) x&-x
using namespace std;
const int maxn=1e5+10,inf=1<<30;
struct node
{
	int x,id;
	bool operator<(const node&t)const
	{
		return x<t.x;
	}
}a[maxn];
struct Q
{
	int x,y,k,id;
}qu[maxn],tmp[maxn];
int c[maxn],ans[maxn],n,o;
void add(int x,int v)
{
	while(x<=n)
	c[x]+=v,x+=low(x);
}
int sum(int x)
{
	int res=0;
	while(x)
	res+=c[x],x-=low(x);
	return res;
}
void solve(int l,int r,int ql,int qr)
{
	int p=ql,q=qr,m=(l+r)/2;
	
	if(l==r)
	{
		for(int i=ql;i<=qr;i++)ans[qu[i].id]=l;
		return;
	}
	while((a[o+1].x<=m&&o<n))++o,add(a[o].id,1);
	while(o&&a[o].x>m)add(a[o].id,-1),--o;
	for(int i=ql;i<=qr;i++)
	{
		int res=sum(qu[i].y)-sum(qu[i].x-1);
		if(res>=qu[i].k)tmp[p++]=qu[i];
		else tmp[q--]=qu[i];
	}
	reverse(tmp+q+1,tmp+qr+1);
	for(int i=ql;i<=qr;i++)qu[i]=tmp[i];
	if(p>ql)solve(l,m,ql,p-1);
	if(q<qr)solve(m+1,r,q+1,qr);
}
int main()
{
	int T,m;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&m);
		for(int i=1;i<=n;i++)
		a[i].id=i,c[i]=0,scanf("%d",&a[i].x);
		sort(a+1,a+1+n);
		for(int i=1;i<=m;i++)
		qu[i].id=i,scanf("%d%d%d",&qu[i].x,&qu[i].y,&qu[i].k);
		o=0;
		solve(-inf,inf,1,m);
		for(int i=1;i<=m;i++)printf("%d\n",ans[i]);
	}
}

猜你喜欢

转载自blog.csdn.net/ccsu_cat/article/details/83188590