Blog·Tanky WooABOUTRSS

HDOJ 1102 Constructing Roads

17 Sep 2010
这篇博客是从旧博客 WordPress 迁移过来,内容可能存在转换异常。

这题太搞了。按照题目意思,这题只需要输入一组数据,而结果确实多组。一组就WA了。

题目传送门: http://acm.hdu.edu.cn/showproblem.php?pid=1102

分析:看到这题给出的都是矩阵形式,就知道了可以用Prim算法求MST。

资料

最小生成树(MST)讲解请看: http://www.wutianqi.com/?p=1284


解答报告:

#include 
using namespace std;
const int INF=1001;

int n, m;
int arcs[101][101];
int dis[101], vis[101];
int _min;

int main()
{
   // freopen("input.txt", "r", stdin);
    while(scanf("%d", &n;) != EOF)
    {
        for(int i=0; i<n; ++i)
            for(int j=0; j<n; ++j)
                scanf("%d", &arcs[i][j]);
        scanf("%d", &m);
        int a, b;
        for(int i=0; i<m; ++i)
        {
            scanf("%d %d", &a, &b);
            arcs[a-1][b-1] = arcs[b-1][a-1] = 0;  // 这一步处理的好
        }
        memset(vis, 0, sizeof(vis));
        for(int i=1; i<n; ++i)
            dis[i] = arcs[0][i];
        vis[0] = 1;
        int sum = 0, rec = 0;
        dis[0] = 0;
        for(int i=1; i<n; ++i)
        {
            _min = INF;
            for(int j=0; j<n; ++j)
                if(!vis[j] && dis[j]<_min)
                {
                    rec = j;
                    _min = dis[j];
                }
            sum += _min;
            vis[rec] = 1;
            for(int j=0; j<n; ++j)
                if(!vis[j] && arcs[rec][j] < dis[j])
                        dis[j] = arcs[rec][j];
            }
        printf("%d\n", sum);
    }
    return 0;

}