HDOJ 1875 畅通工程再续（最小生成树）

畅通工程再续

Problem Description

相信大家都听说一个“百岛湖”的地方吧，百岛湖的居民生活在不同的小岛中，当他们想去其他的小岛时都要通过划小船来实现。现在政府决定大力发展百岛湖，发展首先要解决的问题当然是交通问题，政府决定实现百岛湖的全畅通！经过考察小组RPRush对百岛湖的情况充分了解后，决定在符合条件的小岛间建上桥，所谓符合条件，就是2个小岛之间的距离不能小于10米，也不能大于1000米。当然，为了节省资金，只要求实现任意2个小岛之间有路通即可。其中桥的价格为 100元/米。

 

Input

输入包括多组数据。输入首先包括一个整数T(T <= 200)，代表有T组数据。
每组数据首先是一个整数C(C <= 100),代表小岛的个数，接下来是C组坐标，代表每个小岛的坐标，这些坐标都是 0 <= x, y <= 1000的整数。

 

Output

每组输入数据输出一行，代表建桥的最小花费，结果保留一位小数。如果无法实现工程以达到全部畅通，输出”oh!”.

 

Sample Input

  
  

   
   2
2
10 10
20 20
3
1 1
2 2
1000 1000
  
  

 

Sample Output

  
  

   
   1414.2
oh!
  
  

解题思路：Kruskal或Prim算法，注意距离小于10或大于1000的两个点之间不能连边。（题目好像是稀疏图，Kruskal算法比Prim算法快接近100ms，不过也可能是因为建图的方式不一样。。。）

代码如下：

Kruskal算法

#include <bits/stdc++.h>
#define INF INT_MAX / 10
#define MEM(arr,val) memset(arr,val,sizeof(arr))
#define EPS 1e-8
const int maxn = 10005;
struct edge
{
    int from;
    int to;
    double dis;
};
int par[maxn],rank[maxn],x[maxn],y[maxn];
edge edges[maxn];
int n,m,cnt;
bool cmp(const edge a,const edge b)
{
    return a.dis < b.dis;
}

void init()
{
    for(int i = 0;i < maxn;i++)
        par[i] = i;
    MEM(rank,0);
}

int find(int x)
{
    return par[x] == x ? x : par[x] = find(par[x]);
}

void unite(int x,int y)
{
    x = find(x);
    y = find(y);
    if(x == y)
        return ;
    if(rank[x] < rank[y]){
        par[x] = y;
    }
    else{
        par[y] = x;
        if(rank[x] == rank[y]){
            rank[x]++;
        }
    }
}

bool same(int x,int y)
{
    return find(x) == find(y);
}

double Kruskal()
{
    std::sort(edges,edges + cnt,cmp);
    init();
    double res = 0;
    int k = 0;
    for(int i = 0;i < cnt;i++){
        edge e = edges[i];
        if(!same(e.from,e.to)){
            unite(e.from,e.to);
            res += e.dis;
            k++;
            if(k == n - 1){
                return res;
            }
        }
    }
    return 0;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        for(int i = 0;i < n;i++){
            scanf("%d %d",&x[i],&y[i]);
        }
        cnt = 0;
        for(int i = 0;i < n;i++){
            for(int j = i;j < n;j++){
                double dis = pow(x[i] - x[j],2.0) + pow(y[i] - y[j],2.0);
                if(dis < 100 || 1000000 < dis)
                    continue;
                edges[cnt].from = i;
                edges[cnt].to = j;
                edges[cnt++].dis = sqrt(dis);
            }
        }
        double ans = Kruskal();
        if(ans - EPS < 0)
            printf("oh!\n");
        else
            printf("%.1f\n",ans * 100);
    }
    return 0;
}

Prim算法

#include <bits/stdc++.h>
#define INF INT_MAX / 10
#define MEM(arr,val) memset(arr,val,sizeof(arr))
#define EPS 1e-8
const int maxn = 105;
struct edge
{
    int to;
    double dis;
    edge(int to,double dis)
    {
        this -> to = to;
        this -> dis = dis;
    }
    edge()
    {

    }
    bool operator < (const edge &a)const
    {
        return dis > a.dis;
    }
};
std::priority_queue<edge> que;
std::vector<edge> G[maxn];
int n;
bool vis[maxn];
int x[maxn],y[maxn];
void init()
{
    for(int i = 0;i <= n;i++){
        G[i].clear();
    }
    MEM(vis,0);
}

double Prim()
{
    while(que.size())
        que.pop();
    double res = 0;
    vis[0] = 1;
    int k = 0;
    for(int i = 0;i < (int)G[0].size();i++)
        que.push(G[0][i]);
    while(que.size()){
        edge e = que.top();
        que.pop();
        if(!vis[e.to]){
            vis[e.to] = 1;
            res += e.dis;
            for(int i = 0;i < (int)G[e.to].size();i++){
                que.push(G[e.to][i]);
            }
        }
    }
    return res;
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        init();
        for(int i = 0;i < n;i++){
            scanf("%d %d",&x[i],&y[i]);
        }
        for(int i = 0;i < n;i++){
            for(int j = i;j < n;j++){
                double dis = pow(x[i] - x[j],2.0) + pow(y[i] - y[j],2.0);
                if(dis < 100 || 1000000 < dis)
                    continue;
                G[i].push_back(edge(j,sqrt(dis)));
                G[j].push_back(edge(i,sqrt(dis)));
            }
        }
        double ans = Prim();
        if(ans - EPS < 0)
            printf("oh!");
        else
        printf("%.1f\n",ans * 100);
    }
    return 0;
}