HDOJ4261 Estimation

一道需要用堆初始化的\(DP\)

原题链接

显然对于每一个部分，当\(b[i]\)为\(a\)对于部分的中位数时，差错最小。设\(S(x,y)\)表示\(x\sim y\)这一部分的差错。
\(DP\)的转移方程应该并不难推。
定义\(f[i][j]\)表示前\(i\)个数字分成\(j\)组导致的差错的最小值。

\(\qquad\qquad f[i][j]=\min\limits_{k=0}^{i-1}\{f[i][j],f[k][j-1]+S(k+1,i)\}\)

如果我们直接暴力计算\(S\)，显然会超时，所以我们需要初始化\(S\)，且因为在初始化过程中需要用到动态中位数，所以我们采用一个小根堆和一个大根堆来维护。
我是维护小根堆堆顶作为中位数。
先在小根堆中插入第一个数，定为当前中位数。
然后循环扫到下一个数，若该数比当前小根堆堆顶小，则插入大根堆，否则插入小根堆。
而在插入过程中，必须保证小根堆的大小比大根堆大\(1\)或相等，而在循环的过程中，小根堆堆顶即是当前区间内的中位数。

#include<cstdio>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
const int N = 2010;
const int K = 27;
int f[N][K], a[N], v[N][N];
priority_queue<int>bg;
priority_queue<int, vector<int>, greater<int> >sm;
int re()
{
    int x = 0;
    char c = getchar();
    bool p = 0;
    for (; c<'0' || c>'9'; c = getchar())
        p = (c == '-' || p) ? 1 : 0;
    for (; c >= '0'&&c <= '9'; c = getchar())
        x = x * 10 + (c - '0');
    return p ? -x : x;
}
inline int minn(int x, int y)
{
    return x < y ? x : y;
}
int main()
{
    int i, j, n, m, k, s_sm, s_bg;
    while (1)
    {
        n = re();
        m = re();
        if (!n && !m)
            return 0;
        for (i = 1; i <= n; i++)
            a[i] = re();
        for (i = 1; i <= n; i++)
        {
            while (!sm.empty())
                sm.pop();
            while (!bg.empty())
                bg.pop();
            for (j = i + 1, s_sm = a[i], s_bg = 0, sm.push(a[i]); j <= n; j++)
            {
                if (a[j] < sm.top())
                {
                    bg.push(a[j]);
                    s_bg += a[j];
                }
                else
                {
                    sm.push(a[j]);
                    s_sm += a[j];
                }
                if (sm.size() > bg.size() + 1)
                {
                    bg.push(k = sm.top());
                    s_sm -= k;
                    s_bg += k;
                    sm.pop();
                }
                if (bg.size() > sm.size())
                {
                    sm.push(k = bg.top());
                    s_bg -= k;
                    s_sm += k;
                    bg.pop();
                }
                v[i][j] = s_sm - sm.size()*sm.top() + bg.size()*sm.top() - s_bg;
            }
        }
        memset(f, 60, sizeof(f));
        f[0][0] = 0;
        for (j = 1; j <= m; j++)
            for (i = j; i <= n; i++)
                for (k = 0; k < i; k++)
                    f[i][j] = minn(f[i][j], f[k][j - 1] + v[k + 1][i]);
        printf("%d\n", f[n][m]);
    }
    return 0;
}

ps：因为我用的是\(VS\)，所以代码会自动补空格，而我本来的风格是没有空格的。