省政府“暢通工程”的目標是使全省任何兩個村莊間都可以實現公路交通(但不一定有直接的公路相連,只要能間接通過公路可達即可)。現得到城鎮道路統計表,表中列出了任意兩城鎮間修建道路的費用,以及該道路是否已經修通的狀態。現請你編寫程序,計算出全省暢通需要的最低成本。
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;
}