Kruskal

Kruskal
原理

  1. 把图中的所有边按代价从小到大排序;
  2. 把图中的n个顶点看成独立的n棵树组成的森林;
  3. 按权值从小到大选择边,所选的边连接的两个顶点ui,viui,vi,应属于两颗不同的树,则成为最小生成树的一条边,并将这两颗树合并作为一颗树。
  4. 重复(3),直到所有顶点都在一颗树内或者有n-1条边为止。
    poj1258
#include<iostream>
#include<algorithm>
#include<list>
#include<string.h>
#include<math.h>
#include<vector>
#include<stdio.h>
using namespace std;

#define N 1005

const int Inf =999999;

int pre[N];
struct node{
   int s,e;
   int w;	
}g[N]; 

bool cmp(node a,node b){
	return a.w<b.w;
}
int find(int k){
	while(k!=pre[k]){
		k=pre[k];
	}
	return k;
}

int minTree(int n){
	int total = 0;
	for(int i=0;i<n;i++){
		int s = g[i].s;
		int t = g[i].e;
		if(find(s)!=find(t)){
		 pre[find(s)] = find(t);
		 total += g[i].w;
		}
	}
	return total;
}

int main()
{	
    int n;
    while(cin>>n){
    	memset(g,0,sizeof(g));
    	int k = 0;
    	for(int i=0;i<n;i++){
    		for(int j=0;j<n;j++){
    			int w;
    			cin>>w;
    			if(i==j){
    				continue;
				}else{
					g[k].s = i;
					g[k].e = j;
					g[k].w = w;
					k++;
				}
			}
		}
		for(int i=0;i<n;i++){
			pre[i] = i;  //通过并查集判断是不是已经添加到树上。 
		}
    	sort(g,g+k,cmp); //将权重从小到大排序 
    	int total = minTree(k);
    	cout<<total<<endl;
	}
    
	
    return 0;
    
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章