1. 程式人生 > >poj 1258

poj 1258

pan 一個 tle join rom span -1 main 矩陣

題目:給一個矩陣告訴每條邊的權,求最小生成樹。

題解:建樹之後直接用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;
}
}

poj 1258