acwing 274. 移动服务题解

看着数据范围直接dp即可,注意题意

#include <bits/stdc++.h>
using namespace std;
const int N=205;
const int M=1005;
int dp[M][N][N];//有个人到了第i个指命位子,另外两个分别在j k位子的最小代价.
int a[N][N];
int p[M];
int main()
{
    int n,m;
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            scanf("%d",&a[i][j]);
        }
    }
    for(int i=1;i<=m;i++) scanf("%d",&p[i]);
    p[0]=3;
    memset(dp,0x3f,sizeof dp);
    dp[0][1][2]=0;
    for(int i=0;i<m;i++)
    {
        for(int j=1;j<=n;j++)
        {
            for(int k=1;k<=n;k++)
            {
                int q=p[i];
                int u=p[i+1];
                if (j==k||j==q||k==q) continue;
                dp[i+1][j][k]=min(dp[i+1][j][k],dp[i][j][k]+a[q][u]);
                dp[i+1][q][k]=min(dp[i+1][q][k],dp[i][j][k]+a[j][u]);
                dp[i+1][j][q]=min(dp[i+1][j][q],dp[i][j][k]+a[k][u]);
            }
        }
    }
    int ans=2e9;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            int q = p[m];
            if (j==i||j==q||i==q) continue;
            ans=min(ans,dp[m][i][j]);
        }
    }
    cout<<ans<<endl;
    return 0;
}