HDU-#1863 畅通工程（Kruskal & Prim）

     题目大意：中文题，N个畅通工程了，不解释了....

     解题思路：做了n个畅通版本的题，发现最开始的这个没做，也写了N个MST了，以为一下就可以切了。结果...无限WA，伤心了...一直以为是code敲错了，循环敲了好几次，无果，最后调试才发现是结构体没清空影响的，伤心...

     题目来源：http://acm.hdu.edu.cn/showproblem.php?pid=1863

     Kruskal code:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

const int MAXN = 100+10;
int n,m,tmp;
int p[MAXN];

struct edge{
    int u,v,w;
}e[MAXN*MAXN];

int cmp(const edge a,const edge b){return a.w<b.w;}
int find(int x){return x==p[x]?x:p[x]=find(p[x]);}

int Kruskal(){
    int ans=0;
    for(int i=1;i<=m;i++) p[i]=i;
    sort(e+1,e+1+n,cmp); //下标从1开始的，所以要+1，sort()是从0开始的
    for(int i=1;i<=m;i++){
        int x=find(e[i].u);
        int y=find(e[i].v);
        if(x!=y){
            p[x]=y;
            ans+=e[i].w;
            tmp++; //合并则++，表示有边
        }
    }
    if(tmp==m) return ans; //如果边数计数（初始化为1了，所以++）等于点数，则连通每一个点
    else return 0;
}

int main(){
    while(scanf("%d%d",&n,&m) && n){
        for(int i=1;i<=MAXN;i++){ //清空结构体，一直WA，一直以为是程序的错误，调试了好久才发现是没清空。
            e[i].u=0;e[i].v=0;e[i].w=0;
        }
        for(int i=1;i<=n;i++)
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        tmp=1; //数据完整性计数器
        int ans=Kruskal();
        if(ans) printf("%d\n",ans);
        else printf("?\n");
    }
    return 0;
}

      再补一个Prim算法的，code中要注意初始化范围，不然也很容易出错，导致上次数据遗留。

     Prim code:

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int MAXN = 105;
const int INF =0x3fffffff;
int n,m,a,b,c;
int map[MAXN][MAXN],vis[MAXN],dist[MAXN];

int Prim(){
    int sum=0;
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=MAXN;i++) dist[i] = (i==1 ? 0:map[1][i]);
    vis[1]=1;
    for(int i=1;i<m;i++){
        int pos,minn=INF;
        for(int j=1;j<=m;j++) if(!vis[j] && dist[j]<minn) minn=dist[pos=j];
        if(minn==INF) return 0; //如果有不存在的边，即数据不完整，则返回0
        vis[pos]=1;
        sum+=minn;
        for(int k=1;k<=m;k++)
            if(!vis[k] && dist[k]>map[pos][k])
                dist[k]=map[pos][k];
    }
    return sum;
}

int main(){
    while(scanf("%d%d",&n,&m)!=EOF && n){
        for(int i=0;i<MAXN;i++) //这里初始化一定要初始全部，防止上次数据遗留
            for(int j=0;j<MAXN;j++)
                map[i][j]=INF;
        for(int i=1;i<=n;i++){
            scanf("%d%d%d",&a,&b,&c);
            map[a][b]=map[b][a]=c;
        }
        int ans=Prim();
        if(ans) printf("%d\n",ans);
        else printf("?\n");
    }
    return 0;
}