HDU-1162-Eddy's picture

最新推荐文章于 2019-04-07 18:33:22 发布

f_zyj

最新推荐文章于 2019-04-07 18:33:22 发布

阅读量468

点赞数

分类专栏：树图论文章标签：最小生成树

本文链接：https://blog.csdn.net/f_zyj/article/details/52989932

版权

图论同时被 2 个专栏收录

168 篇文章 0 订阅

订阅专栏

树

75 篇文章 0 订阅

订阅专栏

ACM模版

描述

题解

和HDU-1875-畅通工程再续几乎一模一样啊，最小生成树。
做这两道题时注意到一个小问题，对double cost[][]初始化时不能用mem(cost, 0x3f)，因为是double型，以前从没有注意过这个问题。

代码

#include <iostream>
#include <cstring>
#include <cmath>

#define mem(a, b) memset(a, b, sizeof(a))

using namespace std;

struct land
{
    float x;
    float y;
};

/*
 * Prim求MST
 * 耗费矩阵cost[][]，初始化为INF，标号从0开始，0 ~ n－1
 * 返回最小生成树的权值，返回-1表示原图不连通
 */

const int INF = 0x3f3f3f3f;
const int MAXN = 110;
bool vis[MAXN];
double lowc[MAXN];
double cost[MAXN][MAXN];
struct land L[MAXN];

//  修正cost（添加边）
void updata(int x, int y, double v)
{
    cost[x - 1][y - 1] = v;
    cost[y - 1][x - 1] = v;
    return ;
}

double Prim(double cost[][MAXN], int n)   //  0 ~ n - 1
{
    double ans = 0;
    memset(vis, false, sizeof(vis));
    vis[0] = true;
    for (int i = 1; i < n; i++)
    {
        lowc[i] = cost[0][i];
    }
    for (int i = 1; i < n; i++)
    {
        double minc = INF;
        int p = -1;
        for (int j = 0; j < n; j++)
        {
            if (!vis[j] && minc > lowc[j])
            {
                minc = lowc[j];
                p = j;
            }
        }
        if (minc == INF)
        {
            return -1;  //  原图不连通
        }
        ans += minc;
        vis[p] = true;
        for (int j = 0; j < n; j++)
        {
            if (!vis[j] && lowc[j] > cost[p][j])
            {
                lowc[j] = cost[p][j];
            }
        }
    }
    return ans;
}

int main(int argc, const char * argv[])
{
    int C;

    while (cin >> C)
    {
        //  double不能用mem(cost, 0x3f)来初始化
        for (int i = 0; i < C; i++)
        {
            for (int j = 0; j < C; j++)
            {
                cost[i][j] = INF;
            }
        }
        mem(vis, 0);

        for (int i = 1; i <= C; i++)
        {
            scanf("%f%f", &L[i].x, &L[i].y);
        }

        for (int i = 1; i <= C; i++)
        {
            for (int j = i; j <= C; j++)
            {
                double dis = sqrt(pow(L[i].x - L[j].x, 2) + pow(L[i].y - L[j].y, 2));
                //  cout << res << '\n';
//                if (dis >= 10 && dis <= 1000)
//                {
                    updata(i, j, dis);
//                }
            }
        }

        double ans = Prim(cost, C);
//        if (ans == -1)
//        {
            cout << "oh!\n";
//        }
//        else
//        {
            printf("%.2f\n", ans);
//        }
    }

    return 0;
}