poj2104 K-th Number

http://poj.org/problem?id=2104

整体二分板题，把所有询问按照类似归并排序的方式处理，然后在二分的时候带上值域

如果<=mid的值比k多，那么答案肯定在l,mid之间，否则答案肯定在mid+1到r之间

然后把前一种情况放到前面，后一种情况放到后面，再分别递归子区间。

cdq分治是先递归再合并求，整体二分是先求再递归

#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxl 200010
using namespace std;

int n,m,tot,cnt;
int ans[maxl],a[maxl],b[maxl],num[maxl];
struct que
{
	int l,r,k,id;
}q[maxl],rq[maxl],lq[maxl];

inline void prework()
{
	cnt=0;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		num[i]=a[i];
	}
	sort(num+1,num+1+n);
	tot=unique(num+1,num+1+n)-num-1;
	for(int i=1;i<=n;i++)
	{
		a[i]=lower_bound(num+1,num+1+tot,a[i])-num;
		q[++cnt]=que{i,a[i],0,0};
	}
	que d;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&d.l,&d.r,&d.k);
		d.id=i;q[++cnt]=d;
	}
}

inline void add(int i,int x)
{
	while(i<=n)
	{
		b[i]+=x;
		i+=i&-i;
	}
}

inline int sum(int i)
{
	int ret=0;
	while(i)
	{
		ret+=b[i];
		i-=i&-i;
	}
	return ret;
}

inline void solv(int ql,int qr,int l,int r)
{
	if(ql>qr) return;
	if(l==r)
	{
		for(int i=ql;i<=qr;i++)
		if(q[i].id>0) 
			ans[q[i].id]=l;
		return;
	}
	int mid=(l+r)>>1;
	int lt=0,rt=0;
	for(int i=ql;i<=qr;i++)
	if(q[i].id==0)
	{
		if(q[i].r<=mid)
			add(q[i].l,1),lq[++lt]=q[i];
		else
			rq[++rt]=q[i];
	}
	else
	{
		int d=sum(q[i].r)-sum(q[i].l-1);
		if(d>=q[i].k) 
			lq[++lt]=q[i];
		else
			rq[++rt]=q[i],rq[rt].k-=d;
	}
	for(int i=ql;i<=qr;i++)
	if(q[i].id==0 && q[i].r<=mid)
		add(q[i].l,-1);
	for(int i=1;i<=lt;i++)
		q[ql+i-1]=lq[i];
	for(int i=1;i<=rt;i++)
		q[ql+lt+i-1]=rq[i];
	solv(ql,ql+lt-1,l,mid);
	solv(ql+lt,qr,mid+1,r);
}

inline void mainwork()
{
	for(int i=1;i<=tot;i++)
		b[i]=0;
	solv(1,cnt,1,tot);
}

inline void print()
{
	for(int i=1;i<=m;i++)
		printf("%d\n",num[ans[i]]);
}

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