进阶实验6-3.6 最小生成树的唯一性 (35分)

给定一个带权无向图，如果是连通图，则至少存在一棵最小生成树，有时最小生成树并不唯一。本题就要求你计算最小生成树的总权重，并且判断其是否唯一。

输入格式：
首先第一行给出两个整数：无向图中顶点数 N（≤500）和边数 M。随后 M 行，每行给出一条边的两个端点和权重，格式为“顶点1 顶点2 权重”，其中顶点从 1 到N 编号，权重为正整数。题目保证最小生成树的总权重不会超过 2
​30
​​ 。

输出格式：
如果存在最小生成树，首先在第一行输出其总权重，第二行输出“Yes”，如果此树唯一，否则输出“No”。如果树不存在，则首先在第一行输出“No MST”，第二行输出图的连通集个数。

输入样例 1：
5 7
1 2 6
5 1 1
2 3 4
3 4 3
4 1 7
2 4 2
4 5 5

输出样例 1：
11
Yes

输入样例 2：
4 5
1 2 1
2 3 1
3 4 2
4 1 2
3 1 3

输出样例 2：
4
No

输入样例 3：
5 5
1 2 1
2 3 1
3 4 2
4 1 2
3 1 3

输出样例 3：
No MST
2

后面再看吧，哪里有问题，改不动bug了

#include<iostream>
#include<stdio.h>
#include<vector>
#include<set>
using namespace std;
#define MAX 501
#define INF 1000000
int n,m;
vector<int>v1;
set<int>s1;
int p[MAX][MAX];
int parent[MAX];
int path[MAX];
int dist[MAX];
void prim(int x);
int findmin();
void findroot(int x);
int main(){
    cin>>n>>m;
    for(int i=0;i<=n;i++){
        for(int j=0;j<=n;j++){
            p[i][j]=INF;
        }
    }
    int k1,k2,k3;
    for(int i=1;i<=m;i++){
        scanf("%d %d %d",&k1,&k2,&k3);
        p[k1][k2]=k3;p[k2][k1]=k3;
    }
    prim(1);
}

void prim(int x){
    for(int i=1;i<=n;i++){
        parent[i]=i;
        dist[i]=p[x][i];
    }
    path[1]=x;
    int kk=1;
    dist[x]=0;
    //parent[x]=-1;
    int TotalWeight=0;
    int cnt=1;
    int cnt2=0;
    while(1){
        int newstart=findmin();
        if(newstart==-1){break;}
        cnt2++;
        if(cnt2==1){parent[newstart]=x;}
        path[++kk]=newstart;
        TotalWeight+=dist[newstart];
        //printf("newstart=%d,dist[%d]=%d,TotalWeight=%d\n",newstart,newstart,dist[newstart],TotalWeight);
        dist[newstart]=0;       //这一步很关键，直接把dist[i]=0，那么这个点就不会再参与比较了
        cnt++;
        for(int i=1;i<=n;i++){
            if(dist[i]!=0&&p[newstart][i]<dist[i]){
                    //printf("before:dist[%d]=%d\n",i,dist[i]);
                    dist[i]=p[newstart][i];
                    parent[i]=newstart;
                    /*printf("after:dist[%d]=%d\n",i,dist[i]);
                     for(int i=1;i<=n;i++){
                    //printf("dist[%d]=%d ",i,dist[i]);
                    printf("parent[%d]=%d  ",i,parent[i]);
                }
                printf("\n");*/
                }else if(dist[i]!=0&&p[newstart][i]!=INF&&p[newstart][i]==dist[i]){
                    printf("have the same distance:%d->%d\n",newstart,i);
                    parent[i]=newstart;
                    /*for(int i=1;i<=n;i++){
                        //printf("dist[%d]=%d ",i,dist[i]);
                        printf("parent[%d]=%d  ",i,parent[i]);
                    }*/
                    //printf("\n");
                    v1.push_back(i);
                }
        }
    }
    if(cnt==n){
        printf("%d\n",TotalWeight);
        if(v1.empty()){
            printf("Yes\n");
        }else printf("No\n");
    }else{
         printf("No MST\n");
        for(int i=1;i<=n;i++){
            findroot(i);
        }
        cout<<s1.size()<<endl;
    }
}

int findmin(){
    int index=-1;int minmum=INF;
    for(int i=1;i<=n;i++){
        if(dist[i]!=0&&dist[i]<minmum){
            minmum=dist[i];
            index=i;
        }
    }
    return index;
}

void findroot(int x){
    if(x==parent[x]){
        s1.insert(x);
        //cout<<x<<endl;
        return;
    }
    findroot(parent[x]);
}