算法训练:安慰奶牛

这里写图片描述

解题思路
此题很明显是一个求最小生成树的问题,关键在于如果处理图上边的权值。因为我从七点出发,最终还要沿原路返回起点,所以树上的边皆走过两次。因为每走过一条边都要经过由该边相连的两邻接点,所以最终的权值还要加上两邻接点的权值。一开始我很疑惑点和边同样也要遍历两次为什么不需要乘2。后来发现我们在计算每一条边时都加上了邻接点的权值,所以实际上所有的分支结点都算了两次。通过模拟我们也可以知道在遍历过程中的确只有分支结点会被遍历两次,所以该处理方法是正确的。最后是休息点的问题,只需找到权值最小的结点做休息点即可。综上所述,可以令权值等于两邻接点上权值与边上权值的二倍之和,即e[i].w=p[e[i].u]+p[e[i].v]+2*e[i].w。

代码

#include<iostream>
#include<algorithm>
using namespace std;
struct node
{
    int u,v,w;
};
bool cmp(node a,node b)
{
    return a.w<b.w;
}
int n,m;
node e[100005];
int p[10005],f[10005];
int find(int x)
{
    if(f[x]!=x)
    f[x]=find(f[x]);
    return f[x];
}
int main()
{
    int minl;
    while(cin>>n>>m)
    {
        minl=9999;
        for(int i=1;i<=n;i++)
        {
            cin>>p[i];
            if(minl>p[i])
            minl=p[i];
            f[i]=i;
        }   
        for(int i=1;i<=m;i++)
        {
            cin>>e[i].u>>e[i].v>>e[i].w;
            e[i].w=p[e[i].u]+p[e[i].v]+2*e[i].w;
        }
        sort(e+1,e+1+m,cmp);

        int cnt=0;
        int ans=0;
        for(int i=1;i<=m;i++)
        {
            int u=e[i].u;
            int v=e[i].v;
            int w=e[i].w;
            int t1=find(u);
            int t2=find(v);
            if(t1!=t2)
            {
                ans+=w;
                f[t1]=t2;
                cnt++;
            }
            if(cnt==n-1)
            break;
        }

        ans+=minl;
        cout<<ans<<endl;
    }
    return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值