AcWing 274. 移动服务

原题链接https://www.acwing.com/problem/content/276/

题目描述

一个公司有三个移动服务员，最初分别在位置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

题解

#include "iostream"
#include "algorithm"
#include "cstring"
using namespace std;

const int N = 1010, M = 210;

int f[N][M][M];

int n, m; //m表示位置数，n表示请求数

int p[N];

int w[M][M];

int main(){
    cin>>m>>n;
    
    for(int i = 1; i <= m; i++)
        for(int j = 1; j <= m; j++){
            scanf("%d", &w[i][j]);
        }
    
    for(int i = 1; i <= n; i++) scanf("%d", &p[i]);
    
    memset(f, 0x3f3f3f3f, sizeof(f));
    
    p[0] = 3;
    
    f[0][1][2] = 0;
    
    for(int i = 0; i < n; i++){
        for(int x = 1; x <= m; x++){
            for(int y = 1; y <= m; y++){
                int z = p[i], u = p[i+1], v = f[i][x][y];
                if(x==y||x==z||y==z) continue;
                f[i+1][x][y] = min(f[i+1][x][y], v+w[z][u]); //派z去
                f[i+1][z][y] = min(f[i+1][z][y], v+w[x][u]); //派x去
                f[i+1][x][z] = min(f[i+1][x][z], v+w[y][u]); //派y去
            }
        }
    }
    
    int res = 0x3f3f3f;
    
    for(int x = 1; x <= m; x++){
        for(int y = 1; y <=m ; y++){
            int z = p[n];
            if(x==y||x==z||y==z) continue;
            res = min(res, f[n][x][y]);
        }
    }
    
    printf("%d\n", res);
    
    return 0;
}