poj 1258

Posted 2020-10-03 LMissher

题目：给一个矩阵告诉每条边的权，求最小生成树。

题解：建树之后直接用kruskal。

 

代码：

#include <algorithm>
#include <iostream>
using namespace std;
struct node{//每条边的两顶点以及距离
    int from;
    int to;
    int len;
}edge[5000];

int f[110];
int ans,n,m;

bool cmp(node a, node b){
    return a.len < b.len;
}

int find(int x){//并查集
    while(x!=f[x]){
        x=f[x];
    }
    return x;
}
void join(int x,int y){
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)    f[fx]=fy;
}

int kruskal(){//最小生成树
    sort(edge,edge+m,cmp);//边长度由小到大排序
    ans = 0;
    for(int i=0;i<=n;i++) f[i] = i; for(int j = 0; j < m; j++){
        if(find(edge[j].from) != find(edge[j].to)){//如何两点未在一个并查集便合并
            join(edge[j].from,edge[j].to);
            ans+=edge[j].len;
           //cout<<ans<<endl;
        }
    }
    return ans;
}
int main(){
    while(cin>>n){
        m = n * (n-1) / 2;
        int tot=0,x;
        for(int i=0;i<n;i++)
        for(int j=0;j<n;j++){
        cin>>x;
        if(j<i)
        {
        edge[tot].from=i+1;
        edge[tot].to=j+1;
        edge[tot].len=x;
        tot++;
}
}
        cout<<kruskal()<<endl;
    }
}