HDU1102 Constructing Roads 最小生成树prim||kruskal

题目链接：HDU1102

很裸的题目，下面给出prim算法和kruskal算法解题代码，直接套模板即可。

AC代码：

prim()算法

/*
2017年8月28日15:57:26
HDU1102
最小生成树 
prim算法
AC 
*/
#include <iostream>
//#include <map>
#include <set>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector> 
using namespace std;
const int inf=1<<29;
const int maxn=110;
bool vis[maxn]; 
int dis[maxn]; 
int map[maxn][maxn];
int n,a,b,q; 
void prim(){    
    int i,j,k,tmp,ans=0;    
    for(i=1;i<=n;i++){
    	dis[i]=inf;//初始化     
    	vis[i]=false;
	}    
    	
    dis[1]=0;    
    for(i=1;i<=n;i++){    
        tmp=inf;    
        for(j=1;j<=n;j++){    
            if(!vis[j]&&tmp>dis[j]){    
                tmp=dis[j];    
                k=j;    
            }//找出最小距离的节点     
        } 
		ans+=dis[k];   
        vis[k]=1;//把访问的节点做标记    
        for(j=1;j<=n;j++){    
            if(!vis[j]&&dis[j]>map[k][j])    
            dis[j]=map[k][j];//更新最短距离     
        }    
    }    
    printf("%d\n",ans);
}    

void init(int n){
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&map[i][j]);
		}
	}
	scanf("%d",&q);
	for(int i=1;i<=q;i++){
		scanf("%d%d",&a,&b);
		map[a][b]=map[b][a]=0;
	}
} 

void solve(){
	prim();
}
int main(){
	while(~scanf("%d",&n)){
		init(n);
		solve();
	}
	return 0;
} 

kruskal算法：

/*
2017年8月28日21:05:19
HDU1102
最小生成树
kruskal算法 
AC
*/
#include <iostream>
#include <map>
#include <set>
#include <string>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <vector> 
using namespace std;
const int maxn=1e4+10;
struct node{
	int u,v,w;
	void init(){
		u=v=w=0;
	}
}edge[maxn];
int pre[maxn];
int n,w,num,a,b,q;


void addedge(int u,int v,int w){
	edge[num].u=u;
	edge[num].v=v;
	edge[num].w=w;
	num++; 
} 

bool cmp(node a,node b){
	return a.w<b.w;
}
int find(int x){  
    int r=x;  
    while(pre[r]!=r)    r=pre[r];//查找到根节点为止   
    int i=x,j;  
    while(i!=r){  
        j=pre[i];  
        pre[i]=r;  
        i=j;  
    }  
    return r;   
}

int kruskal(int n){
	int ans=0;
//	int cnt=0;
	sort(edge,edge+num,cmp);
	for(int i=0;i<num;i++){
		int u=edge[i].u;
		int v=edge[i].v;
		int w=edge[i].w;
		u=find(u);
		v=find(v);
		if(u!=v){
			ans+=w;
			pre[v]=u;
		//	cnt++;
		}
	}
	return ans;
}

void init(){
	for(int i=1;i<=maxn;i++) edge[i].init();
	for(int i=1;i<=n;i++) pre[i]=i;
	num=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			scanf("%d",&w);
			if(i>=j) continue;
			else addedge(i,j,w);
		}
	}
	scanf("%d",&q);
	for(int i=1;i<=q;i++){
		scanf("%d%d",&a,&b);
		a=find(a);
		b=find(b);
		pre[b]=a;
	}
}
void solve(){
	printf("%d\n",kruskal(n)); 
}
int main(){

	while(~scanf("%d",&n)){
		init();
		solve();
	}
	return 0;
}