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;
}