杭电ACM1301——Jungle Roads~~最小生成树

这题，简单的最小生成树问题。

只是输入的时候比较麻烦，一开始的N，是村庄的个数，下面N - 1 条信息，一开始的大写字母S和数K，是S村庄有K条路连接，后面是K个村庄以及权值。

处理好了输入的数据，就很简单了。

下面的是AC的代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

class data
{
public:
int from, to, cost;
};

data Road[100];
int n, count1;
int par[30];

int cmp(data &a, data &b)             //权值从小到大排
{
return a.cost < b.cost;
}

int finds(int x)                    //并查集查找函数
{
if(par[x] == x)
return x;
else
return par[x] = finds(par[x]);
}

void unite(int x, int y)            //合并函数
{
x = finds(x);
y = finds(y);
if(x == y)
return;
else
par[y] = x;
}

int same(int x, int y)               //判断是否属于同一个并查集
{
return finds(x) == finds(y);
}

int solve()                         //构造最小生成树函数
{
int res = 0;
for(int i = 0; i < count1; i++)
{
data d = Road[i];
if(!same(d.from, d.to))
{
unite(d.from, d.to);
res += d.cost;
}
}
return res;
}

int main()
{
//	freopen("data.txt", "r", stdin);
int i, j, k, a, w;
char s;
while(scanf("%d", &n) != EOF && n)       //输入的处理
{
getchar();
count1 = 0;
for(i = 0; i < 30; i++)
par[i] = i;
for(i = 0; i < n - 1; i++)
{
scanf("%c%d", &s, &k);
a = s - 'A';
getchar();
for(j = 0; j < k; j++)
{
scanf("%c%d", &s, &w);
Road[count1].from = a; Road[count1].to = s - 'A'; Road[count1].cost = w;
count1++;
getchar();
}
}

sort(Road, Road + count1, cmp);
int ans = solve();
printf("%d\n", ans);
}
return 0;
}