一个公司有三个移动服务员,最初分别在位置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的阶段
执行第i个任务之前第1~i-1任务都执行完了
设f[i][x][y][z]为执行第i个任务时的最小花费,且三人在位置x,y,z.
那么
f[i+1][p][y][z]=min(f[i+1][x][y][z],f[i][x][y][z]+w[x][p])
f[i+1][x][p][z]=min(f[i+1][x][y][z],f[i][x][y][z]+w[y][p])
f[i+1][x][y][p]=min(f[i+1][x][y][z],f[i][x][y][z]+w[z][p])
可惜会超时,要优化。
惊奇的发现
执行第i个任务时三个人中一定有一个人在执行第i-1个任务那里
那么方程优化成如下
f[i][j][k]=min(f[i][j][k],f[i-1][j][k]+c[last][a[i]]);
f[i][a[i]][k]=min(f[i][a[i]][k],f[i-1][j][k]+c[j][a[i]]);
f[i][j][a[i]]=min(f[i][j][a[i]],f[i-1][j][k]+c[k][a[i]]);
上述方程有些奇怪
因为省去的那一维表示的时该阶段执行任务的位置。
我一开始就是不理解这里。
接下来
#include<bits/stdc++.h>
using namespace std;
const int ll=220;
int c[ll][ll],a[1001],f[1001][ll][ll];
int l,n;
int main(){
scanf("%d%d",&l,&n);
for(int i=1;i<=l;i++)
for(int j=1;j<=l;j++)
scanf("%d",&c[i][j]);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
a[0]=3;
memset(f,0x3f,sizeof(f));
f[0][1][2]=0;
for(int i=1;i<=n;i++){
int last=a[i-1];
for(int j=1;j<=l;j++)
for(int k=1;k<=l;k++){
if(j==k||j==last||k==last)
continue;
f[i][j][k]=min(f[i][j][k],f[i-1][j][k]+c[last][a[i]]);
f[i][a[i]][k]=min(f[i][a[i]][k],f[i-1][j][k]+c[j][a[i]]);
f[i][j][a[i]]=min(f[i][j][a[i]],f[i-1][j][k]+c[k][a[i]]);
}
}
int ans=999999999;
for(int i=1;i<=l;i++)
for(int j=1;j<=l;j++)
ans=min(ans,f[n][i][j]);
cout<<ans;
return 0;
}