PAT T1016 Uniqueness of MST

dfs判断连通块的数量，prim算法建立最小生成树并判断是否唯一~

#include<bits/stdc++.h>
using namespace std;
const int maxn=;
const int inf=1e9;
int g[maxn][maxn];
int d[maxn];
int visit[maxn];
int N,M;
struct edge {
    int v1,v2;
};
vector<edge> vi;
void dfs (int s) {
    visit[s]=true;
    for (int i=;i<=N;i++)
    if (g[s][i]!=inf&&visit[i]==false) dfs(i);
}
int dfsTrave () {
    int block=;
    for (int i=;i<=N;i++) {
        if (visit[i]==false) {
            dfs(i);
            block++;
        }
    }
    return block;
}
int flag=;
int prim (int s) {
    fill (d,d+maxn,inf);
    fill (visit,visit+maxn,);
    d[s]=;
    int ans=;
    for (int i=;i<=N;i++) {
        int u=-,min=inf;
        for (int j=;j<=N;j++)
        if (visit[j]==false&&d[j]<min) {
            u=j;
            min=d[j];
        }
        if (u==-) return -;
        visit[u]=;
        ans+=d[u];
        int num=;
        for (int v=;v<=N;v++)
        if (visit[v]&&g[u][v]==min) num++;
        if (num>) flag++;
        for (int v=;v<=N;v++) {
            if (visit[v]==false&&g[u][v]!=inf&&g[u][v]<d[v])
            d[v]=g[u][v];
        }
    }
    return ans;
}
int main () {
    scanf ("%d %d",&N,&M);
    int u,v;
    for (int i=;i<=N;i++)
    for (int j=;j<=N;j++)
    g[i][j]=inf;
    for (int i=;i<M;i++) {
        scanf ("%d %d",&u,&v);
        scanf ("%d",&g[u][v]);
        g[v][u]=g[u][v];
    }
    int block=dfsTrave ();
    if (block>) {
        printf ("No MST\n");
        printf ("%d",block);
        return ;
    }
    fill (visit,visit+maxn,false);
    int mst=prim();
    printf ("%d\n",mst);
    if (flag==) printf ("Yes");
    else printf ("No");
    return ;
}