NEFU OJ 206 最小树（二）

最新推荐文章于 2020-12-12 15:38:39 发布

原创最新推荐文章于 2020-12-12 15:38:39 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

图论同时被 2 个专栏收录

132 篇文章

订阅专栏

贪心

34 篇文章

订阅专栏

本文介绍了一种解决小熊修路问题的方法，通过使用Prim算法来寻找连接所有小熊家的最短路径。文章提供了完整的C++代码实现，并详细解释了如何处理输入和输出数据。

http://acm.nefu.edu.cn/JudgeOnline/contestshow.php?contest_id=120&problem_id=206&num=1

description

在森林里住了n只小熊，他们分别叫小熊A，小熊B……，小熊们决定修建水泥路让他们能更加方便的往来，使得任何一只小熊都能轻松到达其他小熊的家，同时小熊们希望修建的水泥路最短。

input

测试输入若干实例，每个测试实例第一行给出小熊的数目n和小熊们能直接通往的m条道路，（n,m < 100)接下来m行，每行为两只小熊的名字（分别为A,B,C.........,若n为3，则只会出现A,B,C,依次类推）以及这两只小熊之间的距离（为正整数）。

output

输出最短的水泥路的长度，若不能满足任何一只小熊到其他所有小熊的家，则输出-1.

sample_input

3 3
A B 1
A C 2
B C 4
3 1
A B 2

sample_output

3
-1

我的代码：不懂的看上一篇最后的链接

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
#define MAX 99999
#define LEN 101
int dist[LEN];
int map[LEN][LEN];
bool isvisited[LEN];
int m;
void init()//这个函数是用来防止一点到另一点没路时
{
    int i,j;
    for(i=0; i<LEN; i++)
    {
        for(j=0; j<LEN; j++)
        {
            map[i][j]=MAX;
        }
    }
}
int prime(int n)
{
    int i,j,min,pos,sum;
    sum=0;
    memset(isvisited,false,sizeof(isvisited));
    for(i=1; i<=n; i++)
    {
        dist[i]=map[1][i];
    }
    isvisited[1]=true;
    dist[1]=MAX;
    for(i=1; i<n; i++)
    {
        min=MAX;
        for(j=1; j<=n; j++)
        {
            if(!isvisited[j] && dist[j]<min)
            {
                min=dist[j];
                pos=j;
            }
        }
        if(min==MAX)//最小的既然都达到这个程度了，就意味着这两点之间是不连着的
        {
            return -1;
        }
        sum+=min;
        isvisited[pos]=true;
        for(j=1; j<=n; j++)
        {
            if(!isvisited[j] && dist[j]>map[pos][j])
            {
                dist[j]=map[pos][j];
            }
        }
    }
    return sum;
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&m,&n))
    {
        init();
        int i,a,b,w,ans;
        char c,d;
        if(n==0) break;
        for(i=0; i<n; i++)
        {
            cin>>c>>d>>w;//输入若是c语言请看下一个代码片
            a=c-'A'+1;
            b=d-'A'+1;
            if(map[a][b]>w)
            {
                map[a][b]=map[b][a]=w;
            }
        }
        ans=prime(m);
        cout<<ans<<endl;
    }
    return 0;
}

就是输入上有点不同，也粘在这里吧：

#include<iostream>
#include<cstdio>
#include<string.h>
using namespace std;
#define MAX 99999
#define LEN 101
int dist[LEN];
int map[LEN][LEN];
bool isvisited[LEN];
int m;
void init()
{
    int i,j;
    for(i=0; i<LEN; i++)
    {
        for(j=0; j<LEN; j++)
        {
            map[i][j]=MAX;
        }
    }
}
int prime(int n)
{
    int i,j,min,pos,sum;
    sum=0;
    memset(isvisited,false,sizeof(isvisited));
    for(i=1; i<=n; i++)
    {
        dist[i]=map[1][i];
    }
    isvisited[1]=true;
    dist[1]=MAX;
    for(i=1; i<n; i++)
    {
        min=MAX;
        for(j=1; j<=n; j++)
        {
            if(!isvisited[j] && dist[j]<min)
            {
                min=dist[j];
                pos=j;
            }
        }
        if(min==MAX)
        {
            return -1;
        }
        sum+=min;
        isvisited[pos]=true;
        for(j=1; j<=n; j++)
        {
            if(!isvisited[j] && dist[j]>map[pos][j])
            {
                dist[j]=map[pos][j];
            }
        }
    }
    return sum;
}
int main()
{
    int n,m;
    while(~scanf("%d%d%*c",&m,&n))
    {
        init();
        int i,a,b,w,ans;
        char c,d;
        if(n==0) break;
        for(i=0; i<n; i++)
        {
            scanf("%c",&c);
            getchar();
            scanf("%c",&d);
            scanf("%d%*c",&w);
            a=c-'A'+1;
            b=d-'A'+1;
            if(map[a][b]>w)
            {
                map[a][b]=map[b][a]=w;
            }
        }
        ans=prime(m);
        printf("%d\n",ans);
    }
    return 0;
}