題目連接: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;
}