题意:给出n个点,(n-1)*n/2条路,其中有若干条已经建好。求最小生成树。
这题用prim应该效果比较好,但kruskal已经很生疏了,这次给出kruskal模板。
#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 110
#define M 6600
using namespace std;
struct node
{
int u,v,w;
}e[M];
int pre[N],n,cnt;
bool cmp(node a,node b)
{
return a.w<b.w;
}
void add_edge(int u,int v,int w)
{
e[cnt].u=u;
e[cnt].v=v;
e[cnt++].w=w;
}
int findset(int v)
{
while(v!=pre[v]) v=pre[v];
return v;
}
void kruskal()
{
sort(e,e+cnt,cmp);
for(int i=1;i<=n;i++) pre[i]=i;
int t=0,ans=0;
for(int i=0;i<cnt;i++)
{
int u=e[i].u,v=e[i].v,w=e[i].w;
int t1=findset(u);
int t2=findset(v);
if(t1!=t2)
{
ans+=w;
pre[t1]=t2;
t++;
}
if(t==n-1) break;
}
cout<<ans<<endl;
}
int main()
{
while(~scanf("%d",&n)&&n)
{
cnt=0;
int m=(n-1)*n/2;
for(int i=0;i<m;i++)
{
int u,v,w,t;
scanf("%d%d%d%d",&u,&v,&w,&t);
if(t) w=0;
add_edge(u,v,w);
}
kruskal();
}
}