大致思路:
Kruscal求最小生成树用到的知识:图的边的结构体、按边权排序、并查集(不产生回路)
但是这道题仍然需要精心思考,原因是有一些边是“已经存在的”,那么在计算最小生成树成本的时候应该只考虑未修成的边。
要注意两个易错点: ① 多组数据,并查集father数组要初始化,一些全局数组要初始化。 ② 由于有些边已经存在,要先对已存在的边进行并查集的处理,否则在边权排序的时候它排到后面去了,你在生成最小生成树的时候就不晓得还以为没有回路其实有!
注:有可能没有最小生成树,判断方法:看有多少个并查集(father[i]==i),有多个则说明没有建成最小生成树。
AC代码:
#include<iostream>
#include<bits/stdc++.h>
using namespace std;
//就是求最小生成树
struct edge
{
int u,v,w;
}e[1005];
bool cmp(edge a,edge b)
{
if(a.w<b.w)
return true;
else
return false;
}
int cnt=1;
void insert(int a,int b,int w)
{
e[cnt].u=a;
e[cnt].v=b;
e[cnt].w=w;
cnt++;
}
//void insert2(int a,int b,int w)
//{
// insert(a,b,w);
// insert(b,a,w);
//}
int isBuilt[1005][1005];
int father[1005];
int get(int a)
{
if(father[a]==a)
return a;
return get(father[a]);
}
void add(int a,int b)
{
a = get(a);
b = get(b);
if(a!=b)
father[a]=b;
}
int main()
{
int N;
while(scanf("%d",&N)!=EOF)
{
if(N==0)
return 0;
//完全不同的一个案例,重新初始化全局数组
for(int i=0;i<N+1;i++)
father[i]=i;
cnt=1;
for(int i=0;i<N;i++)
for(int j=0;j<N;j++)
isBuilt[i][j]=0;
for(int i=0;i<N*(N-1)/2;i++)
{
int a,b,w,c;
cin>>a>>b>>w>>c;
insert(a,b,w);
isBuilt[a][b]=c;
isBuilt[b][a]=c;
if(c==1) //已经修好的,要把并查集给做好
add(a,b);
}
cnt--;
int count=0; //最小生成树中的边的条数
int ans=0;//需要花多少钱
sort(e+1,e+1+cnt,cmp); //按边权从小到大排列边
for(int i=1;i<=cnt && count<N-1;i++)
{
int a=e[i].u;
int b=e[i].v;
//检查和之前加入的边是否产生回路
if(get(a)==get(b))
continue;
//否则,加入最小生成树
add(a,b);
count++;
//但是要注意 如果已经修建好了就不用算钱了
if(isBuilt[a][b]==0 || isBuilt[b][a]==0)
ans+=e[i].w;
}
cout<<ans<<endl;
}
return 0;
}