图结构练习——最小生成树
Time Limit: 1000MS Memory limit: 65536K
题目描述
有n个城市,其中有些城市之间可以修建公路,修建不同的公路费用是不同的。现在我们想知道,最少花多少钱修公路可以将所有的城市连在一起,使在任意一城市出发,可以到达其他任意的城市。
输入
输入包含多组数据,格式如下。
第一行包括两个整数n m,代表城市个数和可以修建的公路个数。(n <= 100, m <=10000)
剩下m行每行3个正整数a b c,代表城市a 和城市b之间可以修建一条公路,代价为c。
输出
每组输出占一行,仅输出最小花费。
示例输入
3 2 1 2 1 1 3 1 1 0
示例输出
2 0
提示
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <algorithm>
using namespace std;
struct node
{
int u;
int v;
int w;
}b[15000];
int a[150];
bool cmp(struct node a, struct node b)
{
return a.w < b.w;
}
int f(int x)
{
while(x != a[x]){
x = a[x];
}
return x;
}
int mer(int x, int y)
{
int f1, f2;
f1 = f(x);
f2 = f(y);
if(f1 != f2){
a[f1] = f2;
return 1;
}else {
return 0;
}
}
int main()
{
int n, m;
int cnt;
int sum;
while(~scanf("%d %d", &n, &m)){//读入n和m,n表示定点个数,m表示边的条数
cnt = 0;
sum = 0;
int i;
for(i = 0;i < m;i++){
scanf("%d %d %d", &b[i].u, &b[i].v, &b[i].w);
}
sort(b,b+m,cmp);// 按权值从小到大对边进行排序
for(i =1;i <= n;i++){//并查集初始化
a[i] = i;
}
for(i = 0;i < m;i++){ // Kruskal算法的核心,开始从小到大枚举每条边
if(mer(b[i].u,b[i].v)){//判断一条边的两个顶点是否已经连通,即判断是否在同一个集合中,如果连通,选用这条边
sum += b[i].w;
cnt++;
}
if(cnt == n-1){//直到选用了n-1条边之后退出
break;
}
}
printf("%d\n", sum);
}
return 0;
}
prim的最小生成树
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#define inf 0x3f3f3f3f
int dis[150];
int book[150];
int a[150][150];
int main()
{
int n, m;
while(~scanf("%d %d", &n, &m)){
memset(book,0,sizeof(book));
memset(dis,0,sizeof(dis));
memset(a,0,sizeof(a));
int i, j, k;
for(i = 1;i <= n;i++){
for(j = 1;j <= n;j++){
if(i == j){
a[i][j] = 0;
}else {
a[i][j] = inf;
}
}
}
int u, v, w;
for(i = 0;i < m;i++){
scanf("%d %d %d", &u, &v, &w);
if(a[u][v] > w){
a[u][v] = w;
a[v][u] = w;
}
}
for(i = 1;i <= n;i++){//初始化dis数组,这里是1号顶点到各个顶点的初始距离,因为当前生成树中只有1号定点
dis[i] = a[1][i];
}
book[1] = 1;//将1号定点加入到最小生成树中去,这里用book来标记是否加入到生成树
int sum = 0;
int count = 1;
int min;
while(count < n){
min = inf;
for(i = 1;i <= n;i++){
if(book[i] == 0 && dis[i] < min){
min = dis[i];
j = i;
}
}
book[j] = 1;
sum += min;
count++;
for(k = 1;k <= n;k++){ //扫描当前j点所有的边,再以j为中间点,更新生成树到每一个非树顶点的距离
if(book[k] == 0 && dis[k] > a[j][k]){
dis[k] = a[j][k];
}
}
}
printf("%d\n", sum);
}
return 0;
}