hdu 3473 Minimum Sum

划分树的应用，套模板

#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有多少个数到左孩子
__int64 sum[20][MAXN]; // 每层1-i的和

void mBuild(int l, int r, int dep)
{
	if (l == r) 
	{
		sum[dep][l] = node[dep][l];
		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;
		if ( i > l)
			sum[dep][i] = sum[dep][i-1] + node[dep][i];
		else
			sum[dep][i] = node[dep][i];
	}
	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);
}
__int64 res;
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]; // 从a到b进入左孩子的数的个数
	int c = frm_left[dep][a-1] - frm_left[dep][l-1]; // 从l到a-1中进入左孩子的个数
	int d = a-l-c; // 从l到a-1中进入右孩子的个数
	int e = frm_left[dep][b] - frm_left[dep][a-1]; // 从a到b进入左孩子的个数
	int f = b - a + 1 - e; // 从a到b进入右孩子的个数


	if (cnt >= k) // 所求的数全在左孩子
	{
		if (f > 0) // 加上大于中数的部分
		{
			if (d> 0)
				res += sum[dep+1][mid+d+f] - sum[dep+1][mid+d];
			else
				res += sum[dep+1][mid+f];	//	这里要特别注意
		}
		// 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
	{
		if (e > 0)
		{
			if ( c > 0)
				res -= sum[dep+1][l+c+e-1] - sum[dep+1][l+c-1];
			else
				res -= sum[dep+1][l+e-1];
		}
		// 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, t;
	int n, m, A, B;
	int i;
	scanf("%d", &t);
	while (t--)
	{
		scanf("%d", &n);
		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);
			A++; B++;
			res = 0;
			int tp = mFind(A,B,1,n,0,(B-A)/2 + 1);
			if ((B-A)&0x1)
				res -= tp;
			printf("%I64d\n", res);
		}
		puts("");
	}
	return 0;
}