入门经典-最小生成树代码与注释

来源:互联网 发布:c语言从入门到精通txt 编辑:程序博客网 时间:2024/04/29 20:06
#include <iostream>#include <algorithm>using namespace std;int r[100],//边的编号 对之排序    w[100],//对应编号边的权值    s[100],//对应边的起始端点    e[100],//对应边的结束端点    p[100],//并查集    n;     //边的数量,点从0开始int cmp(int i,int j) {return w[i]<w[j];}void read(){    cin>>n;    for(int i=0;i<100;++i)    {        r[i]=i;//初始化边的序号        p[i]=i;//并查集初始,每一条边自成一个连通区域        //代表元为其本身    }    for(int i=0;i<n;++i)    {        cin>>s[i]//端点1           >>e[i]//端点2           >>w[i];//权值    }    sort(r,r+n,cmp);}int find_head(int a) {return p[a]==a?a:p[a]=find_head(p[a]);}int Kruskal(){    int ans=0;    for(int i=0;i<n;++i)    {        int t=r[i];     //取得第i+1小的边的编号        int x=find_head(s[t]);  //取得该边的端点1的代表元        int y=find_head(e[t]);  //取得该边的端点2的代表元        if(x!=y)//不属于同一连通区域        {            ans += w[t];//如未成环加入权值            p[x]=y;  //s(t)所属连通区域与e(t)合并,即代表元相同        }    }    return ans;}int main(){    read();    int ans=Kruskal();    for(int i=0;i<n;++i)        cout<<r[i]<<" "<<s[r[i]]<<" "<<e[r[i]]<<" "<<w[r[i]]<<endl;    cout<<ans;    return 0;}

0 0
原创粉丝点击