POJ 2421

水题,直接贴代码了

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

#define LLEN 105

int map[LLEN][LLEN];
int p[LLEN];

typedef struct Edge {
	int u, v;
	int w;
}Edge;
Edge edge[10050];

int cmp(const void *a, const void *b) {
	Edge aa = *(const Edge *)a;
	Edge bb = *(const Edge *)b;
	return aa.w - bb.w;
}

int find(int x) {
	int s;
	for(s = x; p[s] >= 0; s = p[s]);
	while(s != x) {
		int tmp = p[x];
		p[x] = s;
		x = tmp;
	}
	return s;
}

void init(int x) {
	for(int i = 1; i <= x; i++) {
		p[i] = -1;
	}
}

void uion(int a, int b) {
	int t1 = find(a);
	int t2 = find(b);
	int tmp = p[t1] + p[t2];
	if(p[t1] > p[t2]) {
		p[t1] = t2;
		p[t2] = tmp;
	}
	else {
		p[t2] = t1;
		p[t1] = tmp;
	}
}

int kruskal(int n) {
	int ans = 0;
	for(int i = 0; i < n; i++) {
		int u = edge[i].u;
		int v = edge[i].v;
		if(find(u) != find(v)) {
			uion(u, v);
			ans += edge[i].w;
		}
	}
	return ans;
}

int main() {
	int n, q;
	while(scanf("%d", &n) != EOF) {
		init(n);
		memset(edge, 0, sizeof(edge));
		for(int i = 1; i <= n; i++) {
			for(int j = 1 ; j <= n; j++) {
				scanf("%d", &map[i][j]);
			}
		}
		scanf("%d", &q);
		for(int i = 0; i < q; i++) {
			int a, b;
			scanf("%d%d", &a, &b);
			if(find(a) != find(b)) {
				uion(a, b);
			}
		}
		int k = 0;
		for(int i = 1; i <= n; i++) {
			for(int j = i+1; j <= n; j++) {
				edge[k].u = i;
				edge[k].v = j;
				edge[k++].w = map[i][j];
			}
		}
		qsort(edge, k, sizeof(edge[0]), cmp);
		printf("%d\n", kruskal(k));
	}
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值