POJ 2761 尺取法 + 树状数组

大致题意

Jiajia 要投喂 n 条狗，每条狗都有一个漂亮指数，指数越小代表越漂亮。饭点让狗狗按编号顺序排成一排，投喂 m 次。每次选择 [i, j] 区间的第 k 漂亮的狗喂食。区间可能会存在重叠，但不会存在被另一个区间完全包含的区间。 n<100001, m<50001

即查询区间第 k 大的数值。若用归并树做，查询部分的复杂度和 m 正相关。而基于查询区间的特点，可以使用尺取法降低复杂度。如果求区间第 k 大的值用二分答案，则复杂度为 O(2nlog²n)。

因为不同区间不会被完全包含，所以不存在左边界相等的区间。于是可以将区间按左边界排序（要记录原始顺序再排序，答案需要顺序输出），按顺序查询，每次排除上次查询区间中与本次查询区间不重叠的元素，再加入新的元素。

将狗狗的指数离散化后，用树状数组维护区间内的元素，在新增元素位置加 1，排除元素位置减 1，可以通过判断 [1, t] 区间的求和是否小于 k 来二分答案。

不过，对于树状数组，存在更高效的求解使 x < k 的最右位置的 x 的方法。这与树状数组编号的性质有关，其编号的二进制表示能够与区间容易的对应起来，结尾的 0 的个数对应了不同的区间长度。所以可以按照区间长度由大到小的顺序按照贪心法进行搜索。此时，整个查询的复杂度降为 O(2nlogn)。

#include <cstdio>
#include <STDLIB.H>
#include <algorithm>
#define min(a,b)    (((a) < (b)) ? (a) : (b))
#define max(a,b)    (((a) > (b)) ? (a) : (b))
#define abs(x)    ((x) < 0 ? -(x) : (x))
#define INF 0x3f3f3f3f
#define eps 1e-5
#define M_PI 3.14159265358979323846
#define MAX_M 50005
#define MAX_N 100005
using namespace std;
struct query{
	int l, r, k, id;
	bool operator < (const query& other){
		if(l == other.l) return r < other.r;
		return l < other.l;
	}
};
int N, M;
int X[MAX_N];
int sx[MAX_N], rx[MAX_N], mp[MAX_N + 1];
query qs[MAX_M];
int res[MAX_M];

int bit[MAX_N + 1];
void add(int i, int x){
	while(i <= N){
		bit[i] += x;
		i += i & -i;
	}
}
/* [0, i] 区间求和以二分答案
int sum(int i){
	int s = 0;
	while(i > 0){
		s += bit[i];
		i -= i & -i;
	}
	return s;
}
*/
//按区间长度由大到小顺序搜索，以求得小于 k 的最右位置的值
int find_k(int k){
	int res = 0, sum = 0;
	for(int i = 17; i >= 0; i--){
		res += 1 << i;
		if(res > N || sum + bit[res] >= k) res -= 1 << i;
		else sum += bit[res];
	}
	return res + 1;
}
//数据离散化
bool cmp(const int &a, const int &b){
	return X[a] < X[b];
}
void compress(){
	for(int i = 0; i < N; i++) rx[i] = i;
	sort(rx, rx + N, cmp);
	mp[sx[rx[0]] = 1] = X[rx[0]];
	for(int i = 1; i < N; i++){
		int pre = rx[i - 1], now = rx[i];
		if(X[now] == X[pre]) sx[now] = sx[pre];
		else mp[sx[now] = sx[pre] + 1] = X[now];
	}
}

void init(){
	for(int i = 0; i < N; i++) scanf("%d", X + i);
	compress();
	for(int i = 1; i <= M; i++){
		scanf("%d%d%d", &qs[i].l, &qs[i].r, &qs[i].k);
		--qs[i].l, --qs[i].r;
		qs[i].id = i - 1;
	}
	sort(qs + 1, qs + M + 1);
	qs[0].l = N, qs[0].r = -1;
	memset(bit, 0, sizeof(bit));
}
void solve(){
	for(int i = 1; i <= M; i++){ //尺取法
		int l = max(qs[i - 1].r + 1, qs[i].l), r = min(qs[i - 1].r, qs[i].l - 1);
		for(int j = l; j <= qs[i].r; j++) add(sx[j], 1);
		for(int j = qs[i - 1].l; j <= r; j++) add(sx[j], -1);
		/* 二分答案
		int left = 0, right = sx[rx[N - 1]];
		while(right - left > 1){
			int mid = (left + right) >> 1;
			if(sum(mid) < qs[i].k) left = mid;
			else right = mid;
		}
		res[qs[i].id] = right;
		*/
		res[qs[i].id] = find_k(qs[i].k);
	}
	for(int i = 0; i < M; i++) printf("%d\n", mp[res[i]]);
}
int main(){
	while(~scanf("%d%d", &N, &M)){
		init();
		solve();
	}
	return 0;
}