hdu 4251 The Famous ICPC Team Again

划分树基础

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;
// 划分树，主要解决区间内第几大的数

const int MAXN = 100005;
int node[20][MAXN];// 表示每层上每个位置的值
int num[MAXN]; // 排序的数
int frm_left[20][MAXN]; // 每层1-i有多少个数到左孩子

void mBuild(int l, int r, int dep)
{
	if (l == r) return;
	int mid = (l+r)>>1;
	int same = mid-l+1; // 表示等于中间值且分到左孩子的数的个数
	int i;
	for ( i = l; i<= r; ++i)
	{
		if (node[dep][i] < num[mid])
			--same;
	}
	int ll = l;
	int rr = mid+1;
	
	for (i = l; i<= r; ++i)
	{
		if (node[dep][i] < num[mid]) // 分配给左孩子
		{
			node[dep+1][ll++] = node[dep][i];
		}
		else if (node[dep][i] == num[mid] && same > 0) // 去左边
		{
			node[dep+1][ll++] = node[dep][i];
			--same;
		}
		else	// 右孩子
			node[dep+1][rr++] = node[dep][i];
		frm_left[dep][i] = frm_left[dep][l-1] + ll - l;
	}
	mBuild(l, mid, dep+1);
	mBuild(mid+1, r, dep+1);
}

int mFind( int a, int b, int l, int r, int dep, int k)
{
	if (a == b) return node[dep][a];
	int mid = (l+r)>>1;
	int cnt = frm_left[dep][b] - frm_left[dep][a-1]; // 进入左孩子的数的个数
	if (cnt >= k) // 所求的数全在左孩子
	{
		// L+查询区间去左孩子的数的个数
		int newL =  l + frm_left[dep][a-1] - frm_left[dep][l-1];
		// 左端点+查询区间去左孩子的个数
		int newR = newL + cnt -1;
		return mFind(newL, newR, l, mid, dep+1, k);
	}
	else
	{
		// b + b到R中进入左孩子的个数
		int newR = b + frm_left[dep][r] - frm_left[dep][b];
		// 右端点 - 区间内进入右孩子的个数
		int newL = newR - (b-a-cnt);
		return mFind(newL, newR, mid+1, r, dep+1, k-cnt);
	}
}

int main()   
{
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
#endif
	int cs= 0;
	int n, m, A, B;
	int i;
	while (scanf("%d", &n) == 1)
	{
		printf("Case %d:\n", ++cs);
		for (i = 1; i<= n; i++)
		{
			scanf("%d", &node[0][i]);
			num[i] = node[0][i];
		}
		sort(num+1, num+1+n);	// 构建划分树
		mBuild(1,n,0);
		scanf("%d", &m);
		while (m--)
		{
			scanf("%d%d", &A, &B);
			printf("%d\n", mFind(A,B,1,n,0,(B-A)/2 + 1));
		}
	}
	return 0;
}