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];elsesum[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];elseres += 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];elseres -= 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);#endifint 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;}