longyuxuan789's solution for [CodeForces-295B]

· · 个人记录

#include<bits/stdc++.h>
#define int long long
using namespace std;
int n,dp[505][505],x[505],vis[505],ans[505];
signed main() 
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            cin>>dp[i][j];
        }
    }
    for(int i=1;i<=n;i++)
        cin>>x[i];
    for(int l=n;l>=1;l--)
    {
        int k=x[l];
        vis[x[l]]=1;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                dp[i][j]=min(dp[i][j],dp[i][k]+dp[k][j]);
            }
        }
        int sum=0;
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(vis[i] and vis[j])
                    sum+=dp[i][j];
        ans[l]=sum;
    }
    for(int i=1;i<=n;i++)
        cout<<ans[i]<<" ";
    return 0;
}