I 导航系统

题目 ： 

这个题目里面有一条很重要的话，N条道路用N - 1条边连接起来，这很容易让我们想到最小生成树，那我们就可以把这一条树找出来，那么这连接着的两个点之间的距离也就是他们之间的边，这样我们再通过一个Floyd跑出每两个点之间的最短距离，看看是否和之前的距离一样，如果全部一样，证明可以达到我们的要求输出Yes，然后这些边就是我们通过kruskal时求的那些边，否则输出No

#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
 
const int N = 500 * 500 + 10,M = N*2,INF = 0x3f3f3f3f;
 
int n,m;
int p[N];
int cnt;
const int NN = 510;
 
long long g[NN][NN];
vector<int>v;
struct Edge{
    int a,b,w;
}edge[M];
 
int find(int x)
{
    if(p[x]!=x)p[x] = find(p[x]);
    return p[x];
}
long long d[NN][NN];
 
void kruskal()
{
    sort(edge,edge + cnt,[](Edge a,Edge b){return a.w < b.w;});
    long long res = 0;
    for(int i = 0;i <= n;i++)p[i] = i;
     
    for(int i = 0;i < cnt;i++)
    {
        int a = edge[i].a,b = edge[i].b,w = edge[i].w;
        int fa = find(a),fb = find(b);
        if(fa!=fb)
        {
            d[a][b]= d[b][a] = g[a][b];
            p[fa] = fb;
            res += w;
            v.push_back(w);
        }
    }
}
 
void floyd()
{
    for(int k = 0;k < n;k++)
    {
        for(int i = 0;i< n;i++)
            for(int j = 0;j < n;j++)
                d[i][j] = min(d[i][j],d[i][k]+d[k][j]);
    }
}
 
int main()
{
    cin >> n;
    for(int i = 0;i < n;i++)
    {
        for(int j = 0;j < n;j++)
        {
            cin >> g[i][j];
            if(i!=j)
            edge[cnt++] = {i,j,g[i][j]};
        }
    }
    memset(d,0x3f,sizeof d);
    for(int i = 0;i < n;i++)
    {
        d[i][i] = 0;
    }
     
    kruskal();
     
    /*for(int i = 0;i < n;i++)
    {
        d[i][i] = 0;
        for(int j = 0;j < n;j++)
        {
            cout << d[i][j] << " ";
        }
        cout << endl;
    }*/
     
    floyd();
     
    for(int i = 0;i < n;i++)
    {
        for(int j = 0;j < n;j++)
        {
            if(d[i][j] != g[i][j])
            {
                cout << "No" << endl;
                return 0;
            }
        }
    }
     
    cout << "Yes" << endl;
    for(auto &x : v)
    {
        cout << x << endl;
    }
}