一个公司有三个移动服务员，最初分别在位置1，2，3处。
如果某个位置（用一个整数表示）有一个请求，那么公司必须指派某名员工赶到那个地方去。
某一时刻只有一个员工能移动，且不允许在同样的位置出现两个员工。
从 p 到 q 移动一个员工，需要花费 c(p,q)。
这个函数不一定对称，但保证 c(p,p)=0。
给出N个请求，请求发生的位置分别为 p1~pN。
公司必须按顺序依次满足所有请求，且过程中不能去其他额外的位置，目标是最小化公司花费，请你帮忙计算这个最小花费。
输入格式
第1行有两个整数L,N，其中L是位置数量，N是请求数量，每个位置从1到L编号。
第2至L+1行每行包含L个非负整数，第i+1行的第j个数表示c(i,j) ，并且它小于2000。
最后一行包含N个整数，是请求列表。
一开始三个服务员分别在位置1，2，3。
输出格式
输出一个整数M，表示最小花费。
数据范围
3≤L≤200,
1≤N≤1000
输入样例：
5 9
0 1 1 1 1
1 0 2 3 2
1 1 0 4 1
2 1 5 0 1
4 2 3 4 0
4 2 4 1 5 4 3 2 1
输出样例：
5
方法一：dp
最朴素的做法是：暴搜所有请求，对于每个请求都分别尝试让三个服务员去一次，选最优结果；
暴力dp的做法是：f[i][x][y][z] 表示对于第i个A在x，B在y，C在z的方案数，可是内存不够用，\(200^3 × 2000\) 达到了\(10^10\)，MLE
使用滚动数组优化掉第一为，但不知道哪里不对，求不出答案...
#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int main() {
std::ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,m; cin>>n>>m;
int c[n+1][n+1];
for (int i=1; i<=n; i++)
for (int j=1; j<=n; j++)
cin>>c[i][j];
int A[m+1], f[2][n+1][n+1][n+1]; for (int i=1; i<=m; i++) cin>>A[i]; memset(f, inf, sizeof f);
f[0][1][2][3]=0;
int s=1, ans=inf;
for (int i=0; i<m; i++) {
int& u=A[i+1];
for (int x=1; x<=n; x++)
for (int y=1; y<=n; y++)
for (int z=1; z<=n; z++) if (x!=y && y!=z && f[s^1][x][y][z]!=inf) {
f[s][u][y][z]=min(f[s][u][y][z], f[s^1][x][y][z]+c[x][u]);
f[s][x][u][z]=min(f[s][x][u][z], f[s^1][x][y][z]+c[y][u]);
f[s][x][y][u]=min(f[s][x][y][u], f[s^1][x][y][z]+c[z][u]);
if (i==m-1) ans=min(ans, min(f[s][u][y][z], min(f[s][x][u][z], f[s][x][y][u])));
}
s^=1;
}
cout << ans;
return 0;
}
复杂度分析
- Time：\(O(n × L^3)\)，
- Space：\(O(n × L^3)\)，
方法二：优化 dp
第 i 个订单 p[i] 执行以后一定会有一个服务员在 p[i] 位置，所以我只需要记录两个服务员的位置即可，第三个服务员直接默认为 p[i]，然后状态可压缩为：f[i][x][y]
#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
int main() {
std::ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
int n,m; cin>>n>>m;
int c[n+1][n+1];
for (int i=1; i<=n; i++)
for (int j=1; j<=n; j++)
cin>>c[i][j];
int p[m+1], f[m+1][n+1][n+1]; for (int i=1; i<=m; i++) cin>>p[i]; memset(f, inf, sizeof f);
f[0][1][2]=0; p[0]=3;
int s=1, ans=inf;
for (int i=0; i<m; i++) {
int z=p[i], nx=p[i+1];
for (int x=1; x<=n; x++)
for (int y=1; y<=n; y++) if (x!=y && y!=z && x!=z) {
f[i+1][x][z]=min(f[i+1][x][z], f[i][x][y]+c[y][nx]);
f[i+1][y][z]=min(f[i+1][y][z], f[i][x][y]+c[x][nx]);
f[i+1][x][y]=min(f[i+1][x][y], f[i][x][y]+c[z][nx]);
if (i==m-1) ans=min(ans, min(f[i+1][y][z], min(f[i+1][x][z], f[i+1][x][y])));
}
}
cout << ans;
return 0;
}
复杂度分析
- Time：\(O(nL^2)\)，
- Space：\(O(nL^2)\)，