poj - 1160 - Post Office（dp）

题意：一条直路上有V个村庄(1 <= V <= 300)，现在建P个邮局(1 <= P <= 30)，求每个村庄到其最近邮局的最短距离和的最小值。

题目链接：http://poj.org/problem?id=1160

——>>状态：dp[i][j]表示前i个村庄建j个邮局的最短距离和。

状态转移方程：dp[i][j] = min(dp[i][j], dp[k][j - 1] + nDis[k + 1][i]);（前k个村庄由前j - 1个邮局管，后面的村庄归最后一个邮局管）

状态：nDis[i][j]表示第i个村庄到第j个村庄建1个邮局的最短距离和。

状态转移方程：nDis[i][j] = nDis[i][j-1] + x[j] - x[(i + j) / 2];

总时间复杂度：O(V * V * P)

#include <cstdio>
#include <algorithm>
#include <cstring>

using std::min;

const int MAXV = 300 + 1;
const int MAXP = 30 + 1;

int x[MAXV];
int dp[MAXV][MAXP];
int nDis[MAXV][MAXV];

void Read(int V)
{
    for (int i = 1; i <= V; ++i)
    {
        scanf("%d", x + i);
    }
}

void Init(int V)
{
    memset(nDis, 0, sizeof(nDis));
    for (int i = 1; i <= V; i++)
    {
        for (int j = i; j <= V; j++)
        {
            nDis[i][j] = nDis[i][j-1] + x[j] - x[(i + j) / 2];
        }
    }
}

void Dp(int V, int P)
{
    memset(dp, 0x3f, sizeof(dp));
    for (int i = 0; i <= P; ++i)
    {
        dp[i][i] = 0;
    }
    for (int i = 1; i <= V; ++i)
    {
        // dp[i][i] = 0; 这里初始化会越界，小心！
        for (int j = 1; j <= P && j <= i; ++j)
        {
            for (int k = j - 1; k < i; ++k)
            {
                dp[i][j] = min(dp[i][j], dp[k][j - 1] + nDis[k + 1][i]);
            }
        }
    }
}

void Output(int V, int P)
{
    printf("%d\n", dp[V][P]);
}

int main()
{
    int V, P;

    while (scanf("%d%d", &V, &P) == 2)
    {
        Read(V);
        Init(V);
        Dp(V, P);
        Output(V, P);
    }

    return 0;
}