题目链接<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);
}
}