HDU 2665 整体二分

hdu 2665 传送门

题意：查询区间内第k小的数。

思路：主席树或者整体二分。

学习了一下整体二分，理解的还是不怎么透彻，有一些感觉，推荐文章：整体二分

个人感悟：

整体二分分治思想：

1. 先统计L~R内原始数列中小于等于 二分答案mid 的个数有几个；树状数组，结果为tmp
2. 若查询区间需要的k小于等于tmp， 意味着这tmp个数中一定有它要的那个第k小数，则与原始数列中<=mid的数分在一起，即第一类
3. 若查询区间需要的k大于tmp，意味着这tmp个数不包括其所需，但是都比第k小的数小，所以k减去前tmp小，分在第二类
4. 由此，构成了新的情形，继续二分下去

注意点：

1. 整体二分应用于可二分、离线查询的问题。
2. 记得消除本次二分统计的<=mid的个数，避免对下次二分的影响。

#include <bits/stdc++.h>
#include<cstring>
using namespace std;
const int maxn = 1e5+5, maxm = 2e5+5;
#define lowbit(x) x& -x

struct Que{
	int l, r, k, type, id;
}q[maxm], q1[maxm], q2[maxm];	//依次为原始序列+查询区间；<=mid的暂存区间；>mid的暂存区间 

int T, n, m, ans[maxn], c[maxn];	//第k小答案，树状数组求和数组 
void add(int x, int d){
	while(x <= n){
		c[x] += d;
		x += lowbit(x);
	}
}
int sum(int x){
	int res = 0;
	while(x > 0){
		res += c[x];
		x -= lowbit(x);
	}
	return res;
}

void binary(int L, int R, int l, int r){
	if(L > R) return;
	if(l == r){		//查询到答案 
		for(int i=L; i<=R; ++i)
			if(q[i].type)	ans[q[i].id] = l;
		return;
	}
	
	int mid = (l+r) >> 1, cnt1 = 0, cnt2 = 0;	//cnt1为<=mid的区间标号（个数），cnt2为>mid的 
	for(int i=L, tmp; i<=R; ++i){
		if(q[i].type){	//如果是查询区间，这里注意可能先进行的是下面else部分 
			tmp = sum(q[i].r) - sum(q[i].l - 1);	//区间内k小于mid的个数 
			if(q[i].k <= tmp)	q1[++cnt1] = q[i];	//区间内k小于tmp,即所查找的第k小在<mid的区间里，分在第一类 
			else q2[++cnt2] = q[i], q2[cnt2].k -= tmp;	//反之，所查询的k在>mid的区间里，分在第二类，并且要把前tmp个小的减去 
		}
		else if(q[i].k <= mid){	//k小于等于mid的修改 
			add(q[i].id, 1);	//统计l~r内小于mid的个数
			q1[++ cnt1] = q[i];
		}
		else	q2[++ cnt2] = q[i]; //k大于mid的修改 
	}
	for(int i=1; i<=cnt1; ++i)
		if(!q1[i].type)	add(q1[i].id, -1);	//清除之前统计的个数，避免影响后续二分 
	for(int i=1; i<=cnt1; ++i)	q[L+i-1] = q1[i];	//小于mid的部分放在前cnt1个 
	for(int i=1; i<=cnt2; ++i)	q[L+cnt1+i-1] = q2[i];	//放后面 
	binary(L, L+cnt1-1, l, mid);
	binary(L+cnt1, R, mid+1, r);
}

int main(){
	scanf("%d", &T);
	while(T --){
		memset(c, 0, sizeof(c));
		scanf("%d%d", &n, &m);
		for(int i=1; i<=n; ++i){
			scanf("%d", &q[i].k);
			q[i].id = i;
			q[i].type = 0;
		}
		for(int i=1+n; i<=m+n; ++i){
			scanf("%d%d%d", &q[i].l, &q[i].r, &q[i].k);
			q[i].id = i-n;
			q[i].type = 1;
		}
		binary(1, m+n, 0, INT_MAX);    //INT_MAX取决于输入的数的大小，这个题可以是int以内的任何数
		for(int i=1; i<=m; ++i)
			printf("%d\n", ans[i]);
	}
	return 0;
}