题解【AcWing274】移动服务

题面

非常好的优化 \(\text{DP}\) 状态表示的题目。

首先可以设 \(dp_{i,x,y,z}\) 表示已经做完了前 \(i\) 个请求，现在的 \(3\) 名服务员分别在 \(x\) 、\(y\) 、\(z\) 处的最小花费。

然而这样做时间和空间都会爆炸。

考虑如何优化。

我们注意到做完第 \(i\) 个请求时一定会有一个服务员停留在 \(p_i\) 处，于是可以压掉一维状态。

所以状态就变成了 \(dp_{i,x,y}\) 表示做完了前 \(i\) 个请求，另外 \(2\) 个服务员分别在 \(x\) 和 \(y\) 处的最小花费。

转移的话，枚举下一个请求是由谁做，加上他的位置与下一个请求的位置之间的花费即可。

#include 
#define DEBUG fprintf(stderr, "Passing [%s] line %d\n", __FUNCTION__, __LINE__)
#define itn int
#define gI gi

using namespace std;

typedef long long LL;
typedef pair  PII;
typedef pair  PIII;

inline int gi()
{
    int f = 1, x = 0; char c = getchar();
    while (c < '0' || c > '9') {if (c == '-') f = -1; c = getchar();}
    while (c >= '0' && c <= '9') x = x * 10 + c - '0', c = getchar();
    return f * x;
}

const int maxn = 203, maxm = 1003;

int n, m, l, dp[maxm][maxn][maxn], c[maxm][maxm], p[maxm];

int main()
{
    //freopen(".in", "r", stdin);
    //freopen(".out", "w", stdout);
    l = gi(), n = gi();
    for (int i = 1; i <= l; i+=1)
    {
        for (int j = 1; j <= l; j+=1)
        {
            c[i][j] = gi();
        }
    } 
    for (int i = 1; i <= n; i+=1) p[i] = gi();
    //dp 初始化
    memset(dp, 0x3f, sizeof(dp));
    p[0] = 3;
    dp[0][1][2] = 0;
    for (int i = 0; i < n; i+=1)
    {
        for (int x = 1; x <= l; x+=1)
        {
            for (int y = 1; y <= l; y+=1)
            {
                int z = p[i], nxt = p[i + 1];
                if (x == y || x == z || y == z) continue;
                dp[i + 1][x][y] = min(dp[i + 1][x][y], dp[i][x][y] + c[z][nxt]); //z 去做
                dp[i + 1][z][y] = min(dp[i + 1][z][y], dp[i][x][y] + c[x][nxt]); //x 去做
                dp[i + 1][x][z] = min(dp[i + 1][x][z], dp[i][x][y] + c[y][nxt]); //y 去做
            }
        }
    }
    int ans = 1000000003;
    for (int x = 1; x <= l; x+=1)
    {
        for (int y = 1; y <= l; y+=1)
        {
            int z = p[n];
            if (x == y || y == z || z == x) continue;
            ans = min(ans, dp[n][x][y]);
        }
    }
    printf("%d\n", ans);
    return 0;
}