POJ 1160 四边形不等式优化DP Post Office

d(i, j)表示用i个邮局覆盖前j个村庄所需的最小花费

则有状态转移方程：d(i, j) = min{ d(i-1, k) + w(k+1, j) }

其中w(i, j)的值是可以预处理出来的。

下面是四边形不等式优化的代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;

const int maxp = 30 + 10;
const int maxv = 300 + 10;
const int INF = 0x3f3f3f3f;

int n, m;

int a[maxv], sum[maxv];
int d[maxp][maxv], s[maxp][maxv];

int w(int x, int y)
{
    int t = (x + y) / 2;
    return (t-x) * a[t]-(sum[t-1]-sum[x-1]) + (sum[y]-sum[t])-(y-t)*a[t];
}

int main()
{
    while(scanf("%d%d", &n, &m) == 2)
    {
        for(int i = 1; i <= n; i++) scanf("%d", a + i);
        for(int i = 1; i <= n; i++) sum[i] = sum[i-1] + a[i];

        memset(d, 0x3f, sizeof(d));
        for(int i = 1; i <= n; i++) { d[1][i] = w(1, i); s[1][i] = 0; }
        for(int i = 2; i <= m; i++)
        {
            s[i][n+1] = n;
            for(int j = n; j > i; j--)
            {
                for(int k = s[i-1][j]; k <= s[i][j+1]; k++)
                {
                    if(d[i-1][k] + w(k + 1, j) < d[i][j])
                    {
                        s[i][j] = k;
                        d[i][j] = d[i-1][k] + w(k + 1, j);
                    }
                }
            }
        }
        printf("%d\n", d[m][n]);
    }

    return 0;
}