POJ 1751 High Ways

題目連接: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;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章