POJ11251---Jungle Roads(最小生成树)

【题目来源】https://cn.vjudge.net/problem/POJ-1251
【题意】
这道题很水啦,只不过我新学了prim算法,就拿来练手啦。
这道题题意是在给出的边里选权值和最小的边连通整个村庄。
【思路】
简单的最小生成树模型。
下面先说一下我对prim算法的理解(博主去盗图啦):
这里写图片描述
看这个图,假设我以点D为起点(谁是起点都无所谓的),那么选一条离D点最近的路,当然是DA(方向什么的无所谓。。。),那么现在我手里有两个点,一个是D,一个是A,接下来,就是和最短路求法不一样的地方,我来解释一下,比如迪切斯特拉算法,单源最短路径,因为已经选定了一条边,那么如果是迪切斯特拉算法的话,会拿着这条边去更新其他的边,比如,会比较到底是DB长还是AD+AB长,这样去更新,而prim算法就不一样啦,这个算法只求能够连通,并且用于连接的两点之间是最优的(最省钱的),(最短路求得是一个点到另外一个点的最短路,想一下),所以,prim算法在选出一个点之后,会有这么一个操作:
for(int j=1;j<=n;j++)
{
if(!vis[j]&&d[j]>a[pos][j])
{
d[j]=a[pos][j];
}
}.
再看一下图:
假设已经选出了A点,那么AD高亮显示:
这里写图片描述
这个时候,根据那个代码,会有这么一个操作(只模拟一组)
当更新到B点时,判断一下,是DB短(可以预先赋为无穷大)还是AB更短,根据图片,得到是AB更小,那么d数组里的值就会变化,变得更小。
他不像最短路,非要有一个起点,prim算法只要标记过了一个点,那么这个点和之前的点都是起点,就看哪个更近了。
这里是百度百科:prim算法
【代码一】

#include<cstdio>
#include<cstring>
#include<algorithm>
#define INF 100000
int a[30][30];int n;
int d[30];
bool vis[30];
void prim()
{
    for(int i=1;i<=n;i++)
        d[i]=a[1][i];//初始化,也可以全设为INF
    vis[1]=1;
    int sum=0;
    for(int i=2;i<=n;i++)
    {
        int minn=INF,pos=0;
        for(int j=1;j<=n;j++)//选出一个最近的点
        {
            if(!vis[j]&&d[j]<minn)
            {
                minn=d[pos=j];
            }
        }
        sum+=minn;
        vis[pos]=1;
        for(int j=1;j<=n;j++)//更新
            if(!vis[j]&&d[j]>a[pos][j])
                d[j]=a[pos][j];
    }
    printf("%d\n",sum);
}
int main()
{
    while(~scanf("%d",&n)&&n)
    {
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                a[i][j]=i==j?0:INF;
        getchar();
        for(int i=1; i<n; i++)
        {
            char s,t;
            int num;
            scanf(" %c %d",&s,&num);
            int l=s-'A'+1;
            for(int i=0; i<num; i++)
            {
                int vi;
                scanf(" %c %d",&t,&vi);
                int r=t-'A'+1;
                a[l][r]=a[r][l]=vi;
            }
        }
        prim();
    }
}

【代码二】

#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int pre[30];
struct pp
{
    int x,y,z;
}a[100];
bool cmp(pp c,pp d)
{
    return c.z<d.z;
}
int find(int x)
{
    if(x!=pre[x])
    {
        int fx=find(pre[x]);
        pre[x]=fx;
    }
    return pre[x];
}
int main()
{
    int n;
    while(~scanf("%d",&n)&&n)
    {
        for(int i=1; i<=n; i++)
            pre[i]=i;
        int l=0;
        for(int i=1; i<n; i++)
        {
            char s;
            int num;
            scanf(" %c%d",&s,&num);
            int st=s-'A'+1;
            for(int j=0;j<num;j++)
            {
                char t;
                scanf(" %c%d",&t,&a[l].z);
                a[l].x=st;
                a[l].y=t-'A'+1;
                l++;
            }
        }
        sort(a,a+l,cmp);
        int num=0,money=0;
        for(int i=0;i<l;i++)
        {
            int fa=find(a[i].x);
            int fb=find(a[i].y);
            if(fa!=fb)
            {
                num++;
                money+=a[i].z;
                if(num>=n-1)
                    break;
                pre[fa]=fb;
            }
        }
        printf("%d\n",money);
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值