P2910 [USACO08OPEN] Clear And Present Danger S （Floyd算法）

题目链接：https://www.luogu.com.cn/problem/P2910

题解

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long ll;
int t,n;
int m;
int step[10001];
ll dis[101][101];

int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin>>n>>m;
    // for(int i=1;i<=n;i++)
    // {
    //     for(int j=1;j<=n;j++)
    //     {
    //         dis[i][j]=INT_MAX;
    //     }
    // }
    for(int i=1;i<=m;i++)
    {
        cin>>step[i];
    }
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>dis[i][j];
        }
    }

    for(int k=1;k<=n;k++)
    {
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
            }
        }
    }
    ll ans=0;
    for(int i=1;i<m;i++)
    {
        ans+=dis[step[i]][step[i+1]];
    }
    cout<<ans<<endl;

    return 0;
}