HDU 1863 畅通工程 最小生成树问题 （Kruskal&Prim）

题目链接<http://acm.hdu.edu.cn/showproblem.php?pid=1863>

---

Kruskal：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=1e2+7;
struct Edge{
    int u,v,w;
    bool operator<(const Edge a)const{
        return w<a.w;
    }
}e[N];
int fa[N];
int fd(int x){return x==fa[x]?x:fa[x]=fd(fa[x]);}
int Kruskal(int n,int m){
    for(int i=1;i<=n;i++) fa[i]=i;
    sort(e+1,e+1+m);
    int res=0;
    for(int i=1;i<=m;i++){
        int u=e[i].u,v=e[i].v;
        int fu=fd(u),fv=fd(v);
        if(fu==fv) continue;
        fa[fu]=fv;
        res+=e[i].w;
    }
    for(int i=2;i<=n;i++){
        if(fd(i)!=fd(1)) return -1;
    }
    return res;
}
int main(){
    int n,m;
    while(scanf("%d%d",&m,&n)!=EOF){
        if(m==0) break;
        for(int i=1;i<=m;i++)
            scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
        int ans=Kruskal(n,m);
        if(ans==-1) printf("?\n");
        else printf("%d\n",ans);
    }
}

---

Prim：

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=1e5+7;
const int inf=1e9+7;
struct Edge{
    int u,v,w,nxt;
    Edge(int u=0,int v=0,int w=0,int nxt=0):u(u),v(v),w(w),nxt(nxt){}
}e[N*2];
int p[N],edn;
void add(int u,int v,int w){
    e[++edn]=Edge(u,v,w,p[u]);p[u]=edn;
    e[++edn]=Edge(v,u,w,p[v]);p[v]=edn;
}
int low[N],vis[N];
int Prim(int n,int m){
    int res=0;
    for(int i=1;i<=n;i++) vis[i]=0,low[i]=inf;
    for(int i=p[1];~i;i=e[i].nxt)
        low[e[i].v]=min(low[e[i].v],e[i].w);
    vis[1]=1,low[1]=0;
    for(int t=1;t<n;t++){
        int minn=inf,u=0;
        for(int i=1;i<=n;i++){
            if(!vis[i]&&low[i]<minn)
                minn=low[i],u=i;
        }
        if(!u) return -1;
        vis[u]=1;res+=minn;
        for(int i=p[u];~i;i=e[i].nxt){
            int v=e[i].v;
            if(vis[v]) continue;
            low[v]=min(low[v],e[i].w);
        }
    }
    return res;
}
int main(){
    int n,m;
    while(scanf("%d%d",&m,&n)!=EOF){
        if(m==0) break;
        int u,v,w;
        memset(p,-1,sizeof(p));edn=-1;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        int ans=Prim(n,m);
        if(ans==-1) printf("?\n");
        else printf("%d\n",ans);
    }
}