toj 4353 Estimation(树状数组+二分查找)

Estimation

时间限制(普通/Java):5000MS/15000MS     运行内存限制:65536KByte
总提交: 6            测试通过: 1

描述

 

“There are too many numbers here!” your boss bellows. “How am I supposed to make sense of all of this? Pare it down! Estimate!”
You are disappointed. It took a lot of work to generate those numbers. But, you’ll do what your boss asks.
You decide to estimate in the following way: You have an array A of numbers. You will partition it into k contiguous sections, which won’t necessarily be of the same size. Then, you’ll use a single number to estimate an entire section. In other words, for your array A of size n, you want to create another array B of size n, which has k contiguous sections. If i and j are in the same section, then B[i]=B[j]. You want to minimize the error, expressed as the sum of the absolute values of the differences (Σ|A[i]-B[i]|).

输入

 

There will be several test cases in the input. Each test case will begin with two integers on a line, n (1≤n≤2,000) and k (1≤k≤25, k≤n), where nis the size of the array, and k is the number of contiguous sections to use in estimation. The array A will be on the next n lines, one integer per line. Each integer element of A will be in the range from -10,000 to 10,000, inclusive. The input will end with a line with two 0s.

输出

 

For each test case, output a single integer on its own line, which is the minimum error you can achieve. Output no extra spaces, and do not separate answers with blank lines. All possible inputs yield answers which will fit in a signed 64-bit integer.
 

样例输入

 

7 2
6
5
4
3
2
1
7
0 0

样例输出

 

9

题目来源

The University of Chicago Invitational Programming Contest 2012

题解：

勉强卡过。。应该不是正解。。

以前做过一道题。。在一个线段上给定几个点。。求一个点使得到各点的距离和最短。。。从小到大排序。。该点就是中间大的点。。偶数个的话。。中间两个点都可以

dp[i][j]记录i到j最小的绝对值。。。

树状数组t[i]用来记录大小为i的点的个数

树状数组e[i]用来记录<=i的所有点的和

求i到j这一段的中间值就是找t[mid] == (j - i + 1 + 1) / 2 偶数个的话。。取后一个。。

之后找比 mid 小的所有点和sum1。。还有比 mid 大的所有点和sum2

求比 mid 小的所有点的个数num1。。比 mid 大的点个数num2

dp[i][j] = sum2 - sum1 + mid * (num1 - num2)

 

#include <stdio.h>
#include <string.h>
const int N = 2005;
const int Max = 20010;
int a[N];
int t[Max+10];
int e[Max+10];
int dp[N][N];
int f[N][30];
int Min(int a, int b){
    return a < b ? a : b;
}

int lowbit(int x){
    return x & (-x);
}

void add(int i, int val, int c[]){
    
    while(i <= Max){
        c[i] += val;
        i += lowbit(i);
    }
}
int sum(int i, int c[]){
    
    int ret = 0;
    while(i > 0){
        ret += c[i];
        i -= lowbit(i);
    }
    return ret;
}
int Bin_search(int v){
    
    int l = 1;
    int r = Max;
    while(l < r){
        int m = (l + r) >> 1;
        int p = sum(m, t);
        if(p >= v){
            r = m;
        }
        else{
            l = m + 1;
        }
    }
    return l;
}
int main()
{
    int n, k;
    while(scanf("%d %d", &n, &k), n || k){
        int i, j, g;
        for(i = 1; i <= n; i++){
            scanf("%d", a+i);
            a[i] += 10005;
        }
        for(i = 1; i <= n; i++){
            
            memset(t, 0, sizeof(t));
            memset(e, 0, sizeof(e));
            for(j = i; j <= n; j++){
                add(a[j], 1, t);
                add(a[j], a[j], e);
                int num = (j - i + 2) >> 1;
                int mid = Bin_search(num);
            //    printf("mid = %d
", mid);
                int sum1 = sum(mid - 1, e);
            //    printf("sum1 = %d
", sum1);
                int sum2 = sum(Max, e) - sum(mid, e);
            //    printf("sum2 = %d
", sum2);
                int num1 = sum(mid - 1, t);
            //    printf("num1 = %d
", num1);
                int num2 = sum(Max, t) - sum(mid, t);
            //    printf("num2 = %d
", num2);
                dp[i][j] = sum2 - sum1 + mid * (num1 - num2);
            //    printf("%d ", dp[i][j]);
            }
        }
        for(i = 1; i <= k; i++){
            for(j = 1; j <= n; j++){
                f[j][i] = dp[1][j];
                if(i == 1){
                    continue;
                }
                for(g = 1; g <= j; g++){
                    f[j][i] = Min(f[j][i], f[g-1][i-1] + dp[g][j]);        
                }
                //printf("%d ", f[j][i]);
            }
            //puts("");
        }
        printf("%d
", f[n][k]);
    }
    return 0;
}