HDU 1233 还是畅通工程（Prim，Kruscal）

题目描述：

 

 

 

思路分析：

最小生成树

 

最小生成树——Kruscal

#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <queue>
#include <stack>
#include <string.h>
#include <math.h>
#include <algorithm>

#define INF 0x3f3f3f3f

using namespace std;

int n;
int tree[110];
int g[110][110];

int find(int i){
    if(tree[i]!=-1)return tree[i]=find(tree[i]);
    else return i;
}

void Kruscal(){
    int x,y,ans=0,cnt=n-1;
    memset(tree,-1,sizeof(tree));
    while(cnt){
        int l=INF;
        //printf("%d ",cnt);
        for(int i=1;i<=n;i++){
            for(int j=i+1;j<=n;j++){
                if(g[i][j]<l){
                    l=g[i][j];
                    x=i,y=j;
                }
            }
        }
        //printf("%d %d*\n",x,y);
        int rx=find(x);
        int ry=find(y);
        //printf("%d  %d--\n",rx,ry);
        if(rx!=ry){
            tree[ry]=rx;
            ans+=g[x][y];
            cnt--;
        }
        g[x][y]=INF;
    }
    printf("%d\n",ans);
}

int main(){
    while(scanf("%d",&n),n){
        int t=n*(n-1)/2;
        while(t--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            g[a][b]=c;   //每次寻找权值最短的边无所谓方向，只存储一半即可
        }
        Kruscal();
    }
}

 

最小生成树——Prim（类似与Dijkstra)

#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <queue>
#include <stack>
#include <string.h>
#include <math.h>
#include <algorithm>

#define INF 0x3f3f3f3f

using namespace std;

int n;
int dis[110];
int mark[110];
int g[110][110];

void Prim(){
    memset(dis,0x3f,sizeof(dis));
    memset(mark,0,sizeof(mark));
    int ans=0;
    dis[1]=0;
    for(int i=0;i<n;i++){
        int index,p=INF;
        for(int j=1;j<=n;j++){
            if(!mark[j]&&dis[j]<p){
                p=dis[j];
                index=j;
            }
        }
        ans+=p;
        mark[index]=1;
        for(int j=1;j<=n;j++){
            if(!mark[j]){
                dis[j]=min(dis[j],g[index][j]);
            }
        }
    }
    printf("%d\n",ans);
}

int main(){
    while(scanf("%d",&n),n){
        int t=n*(n-1)/2;
        while(t--){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            g[a][b]=g[b][a]=c;   //松弛操作可以从每条边的任意方向，所以两边都要存
        }
        Prim();
    }
}