原题链接:http://acm.hdu.edu.cn/showproblem.php?pid=1879
样例:
Sample Input
3
1 2 1 0
1 3 2 0
2 3 4 0
3
1 2 1 0
1 3 2 0
2 3 4 1
3
1 2 1 0
1 3 2 1
2 3 4 1
0
Sample Output
3
1
0
题意: 给定n个村庄数目,以及村庄之间的信息,问连通这些村庄的最小成本。
解题思路: 这道题给出的信息为边,所以我们用Kruskal算法来写。对于输入数据要注意的是有一个标志flag表示其中两个村庄是否修建。如果已经修建我们要将之间的消耗成本更改为0。其他的和Kruskal算法模板一样,若对Kruskal算法不填熟悉的话,这里指路一篇blog。
AC代码:
/*
*
*
*/
//POJ不支持
//i为循环变量,a为初始值,n为界限值,递增
//i为循环变量, a为初始值,n为界限值,递减。
using namespace std;
const int inf = 0x3f3f3f3f;//无穷大
const int maxn = 1e5;//最大值。
typedef long long ll;
typedef long double ld;
typedef pair<ll, ll> pll;
typedef pair<int, int> pii;
//*******************************分割线,以上为自定义代码模板***************************************//
struct Edge{
int u,v;//两个村庄的编号。
int w;//两个村庄之间的成本。
bool operator <(const Edge &a){
return w<a.w;//运算符重载。
}
};
Edge edges[maxn];
int verx[maxn];//verx[i]代表i村庄所属的连通分量。
int n,cnt;//n个村庄,cnt代表边数。
int u,v,w;//临时变量。
bool flag;
void Kruskal(){
int sum=0;
rep(i,1,n)verx[i]=i;//初始连通分量为它自己。
sort(edges,edges+cnt);
rep(i,0,cnt-1){
u=verx[edges[i].u];
v=verx[edges[i].v];
if(u!=v){
rep(j,1,n){
if(verx[j]==v)verx[j]=u;
}
sum+=edges[i].w;
}
}
cout<<sum<<endl;
}
int main(){
//freopen("in.txt", "r", stdin);//提交的时候要注释掉
IOS;
while(cin>>n&&n){
cnt=n*(n-1)/2;
rep(i,0,cnt-1){
cin>>u>>v>>w>>flag;
edges[i].u=u;edges[i].v=v;edges[i].w=w;
if(flag){
edges[i].w=0;
}
}
Kruskal();
}
return 0;
}