Hdu3473 Minimum Sum(划分树)

题意：给一个数列x0，x1，，，xn，之后给m次询问（l，r），求使的值最小的x，输出结果。

思路：想了半天，最后确定了x一定是数列的中位数，证明：

对于上图中的情况，四段长度分别为a、b、c、d。

取x为中位数x3时，结果S1=(a+b) + b + 0 + c + (c+d).

对于第二个图，假设取x=x3-△x，得S2=（a+b-△x) + (b-△x) + △x + (c+△x) + (c+d+△x)

       =S1+ 3*△x -2*△x 

      = S1 + △x > S1.

结论得证。长度为偶数的区间证明方法也差不多，中位数取中间的任意一个即可。

有了上面的结论，这题就成了大于中位数的数之和减去小于中位数的数之和，划分树搞一搞即可，多维护一个sum值，记录到当前位置进入左子树的值的和。

case后面少打了个‘#’WA了三天，很崩溃 T_T 。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int mod = 1000000007;
const int maxn=100005;
int tree[20][maxn], toLeft[20][maxn], sorted[maxn];
ll lsum[20][maxn], sum[maxn], tot;
void build(int level, int l, int r){
    if(l == r)return;
    int mid = (l + r) >> 1;
    int cnt = mid - l + 1;
    for(int i=l; i<=r; ++i){
        if(tree[level][i] < sorted[mid]){
            --cnt;
        }
    }
    int lpos = l, rpos = mid + 1;
    for(int i=l; i<=r; ++i){
        if(i == l){
            toLeft[level][i] = 0;
            lsum[level][i] = 0;
        } else {
            toLeft[level][i] = toLeft[level][i - 1];
            lsum[level][i] = lsum[level][i - 1];
        }
        if(tree[level][i] < sorted[mid]){
            tree[level + 1][lpos++] = tree[level][i];
            ++toLeft[level][i];
            lsum[level][i] += tree[level][i];
        } else if(tree[level][i] > sorted[mid]) {
            tree[level + 1][rpos++] = tree[level][i];
        } else {
            if(cnt){
                tree[level + 1][lpos++] = tree[level][i];
                ++toLeft[level][i];
                lsum[level][i] += tree[level][i];
                --cnt;
            } else {
                tree[level + 1][rpos++] = tree[level][i];
            }
        }
    }
    build(level + 1, l, mid);
    build(level + 1, mid + 1, r);
}
int query(int level, int l, int r, int ql, int qr, int k){
    if(ql == qr){
        return tree[level][ql];
    }
    int s1 = (l == ql ? 0 : toLeft[level][ql - 1]);
    int s2 = toLeft[level][qr] - s1;
    ll s = (l == ql ? lsum[level][qr] : lsum[level][qr] - lsum[level][ql - 1]);
    int mid = (l + r) >> 1;
    if(s2 >= k){
        return query(level + 1, l, mid, l + s1, l + s1 + s2 - 1, k);
    } else {
        tot += s;
        return query(level + 1, mid + 1, r, mid + ql - l - s1 + 1, mid - l + qr - s1 - s2 + 1, k - s2);
    }
}
int main(){
    int n, m, l, r, k, H, t, cas = 0;
    scanf("%d", &t);
    while(t--){
        scanf("%d", &n);
        sum[0] = 0;
        for(int i=1; i<=n; ++i){
            scanf("%d", &tree[0][ i]);
            sorted[i] = tree[0][i];
            sum[i] = tree[0][i] + sum[i - 1];
        }
        sort(sorted + 1, sorted + n + 1);
        build(0, 1, n);
        scanf("%d", &m);
        printf("Case #%d:\n", ++cas);
        while(m--){
            scanf("%d%d", &l, &r);
            k = (r - l) >> 1;
            ++k;
            ++l;
            ++r;
            tot = 0ll;
            int mid = query(0, 1, n, l, r, k);
            ll ans = sum[r] - sum[l - 1] - tot - tot - (ll)mid;
            if((r - l) & 1){//区间长度偶数时再减一次中位数
                ans -= (ll)mid;
            }
            printf("%lld\n", ans);

        }
        printf("\n");
    }
    return 0;
}