省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可)。现得到城镇道路统计表,表中列出了任意两城镇间修建道路的费用,以及该道路是否已经修通的状态。现请你编写程序,计算出全省畅通需要的最低成本。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 );随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态,每行给4个正整数,分别是两个村庄的编号(从1编号到N),此两村庄间道路的成本,以及修建状态:1表示已建,0表示未建。
当N为0时输入结束。
Output
每个测试用例的输出占一行,输出全省畅通需要的最低成本。
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
思路:
kruskal模板题,对于已经建好的道路,我的思路就是将建好道路的节点添加到并查集中。
AC代码:
#include<cstdio>
using namespace std;
typedef struct
{
int u;
int v;
int w;
} edge;
void quicksort(edge a[], int left, int right)
{
int i, j, temp;
edge t;
i = left;
j = right;
temp = a[i].w;
if(i > j)
{
return;
}
while(i < j)
{
while(a[j].w >= temp && i < j)
j--;
while(a[i].w <= temp && i < j)
i++;
if(i < j)
{
t = a[i];
a[i] = a[j];
a[j] = t;
}
}
t = a[i];
a[i] = a[left];
a[left] = t;
quicksort(a, left, i-1);
quicksort(a, j+1, right);
}
void init(int a[], int n)
{
int i;
for(i = 1; i <= n; i++)
{
a[i] = i;
}
}
int getf(int a[], int v)
{
if(a[v] == v)
{
return v;
}
else
{
a[v] = getf(a, a[v]);
return a[v];
}
}
int merge(int a[], int u, int v)
{
int t1 = getf(a, u);
int t2 = getf(a, v);
if(t1 == t2)
{
return 0;
}
else
{
a[t2] = t1;
return 1;
}
}
int main()
{
int n, m;
while(scanf("%d", &n) != EOF && n != 0)
{
if(n == 0)
{
break;
}
int a[105], count = 0, sum = 0;
int i, tt;
edge e[6005];
init(a, 104);
n = n*(n-1)/2;
for(i = 1; i <= n; i++)
{
scanf("%d%d%d", &e[i].u, &e[i].v, &e[i].w);
scanf("%d", &tt);
if(tt == 1)
{
merge(a, e[i].u, e[i].v);
}
}
quicksort(e, 1, n);
for(i = 1; i <= n; i++)
{
if(merge(a, e[i].u, e[i].v))
{
count++;
sum += e[i].w;
}
if(count == m-1)
{
break;
}
}
printf("%d\n", sum);
}
return 0;
}