题目连接:http://poj.org/problem?id=1751
题意:
有n个村庄,下面n行是每个村庄的坐标,距离就是坐标之间的距离,要修路保证每个之间是连通的。输入m,下面m行每行输入的是已经连通的村庄的序号。
思路:Kruskal算法:将所有路径按大小排序,依次加入边集中。注意形成环路的时候,这条边不能加入。用并查集去判断是否形成环路。
代码:有RE错误,没有发现错在哪里,求大佬看到指点
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <string>
#include<cstring>
#include <map>
#include <vector>
#include <set>
#include <queue>
#include <stack>
#include <cctype>
using namespace std;
#define Fill(x,a) memset(x,a,sizeof(x));
const int maxn = 760;
const int maxm = 50000;
int F[maxn];
struct point {
int x, y;
}p[maxn];
struct Edge
{
int u, v, w;
}edge[maxm];
int tol;
void addedge(int u, int v, int w)
{
edge[tol].u = u;
edge[tol].v = v;
edge[tol++].w = w;
}
bool cmp(Edge a, Edge b)
{
return a.w<b.w;
}
int find(int x)
{
if (F[x] == -1) return x;
return F[x] = find(F[x]);
}
void Kruskal(int n)
{
Fill(F, -1);
sort(edge, edge + tol, cmp);
int cnt = 0;
int ans = 0;
for (int i = 0; i<tol; i++)
{
int u = edge[i].u;
int v = edge[i].v;
int w = edge[i].w;
int t1 = find(u);
int t2 = find(v);
if (t1 != t2)
{
if (edge[i].w>0)
{
cout << u << " " << v << endl;
}
ans += w;
F[t1] = t2;
cnt++;
}
if (cnt == n - 1) break;
}
}
int getdist(int a, int b)//直接返回的平方值,因为本题不需要输出最后的权值
{
return (p[a].x - p[b].x)*(p[a].x - p[b].x) + (p[a].y - p[b].y)*(p[a].y - p[b].y);
}
int main()
{
//input;
int i, n, m, x, y, z, ans, t, j, k;
scanf("%d", &n) == 1;//n个城市,n个点集
for (i = 1; i <= n; i++)
{
cin >> p[i].x >> p[i].y; //代表第i个点的坐标
}
ans = tol = 0;
//!!!!!!因为i,j就代表的前一个和后一个点,所以i,j从1开始
for (i = 1; i <= n; i++)
{
for (j = i+1; j <= n; j++)
{
addedge(i, j, getdist(i, j));
}
}
cin >> m;
//将已经连通的点之间的权值更新为0
for (i = 1; i <= m; i++)
{
int u, v;
cin >> u >> v;
addedge(u, v, 0);
}
Kruskal(n);
return 0;
}