算法训练 安慰奶牛

问题描述
Farmer John变得非常懒,他不想再继续维护供奶牛之间供通行的道路。道路被用来连接N个牧场,牧场被连续地编号为1到N。每一个牧场都是一个奶牛的家。FJ计划除去P条道路中尽可能多的道路,但是还要保持牧场之间 的连通性。你首先要决定那些道路是需要保留的N-1条道路。第j条双向道路连接了牧场Sj和Ej(1 <= Sj <= N; 1 <= Ej <= N; Sj != Ej),而且走完它需要Lj的时间。没有两个牧场是被一条以上的道路所连接。奶牛们非常伤心,因为她们的交通系统被削减了。你需要到每一个奶牛的住处去安慰她们。每次你到达第i个牧场的时候(即使你已经到过),你必须花去Ci的时间和奶牛交谈。你每个晚上都会在同一个牧场(这是供你选择的)过夜,直到奶牛们都从悲伤中缓过神来。在早上 起来和晚上回去睡觉的时候,你都需要和在你睡觉的牧场的奶牛交谈一次。这样你才能完成你的 交谈任务。假设Farmer John采纳了你的建议,请计算出使所有奶牛都被安慰的最少时间。

输入格式
第1行包含两个整数N和P。

接下来N行,每行包含一个整数Ci。

接下来P行,每行包含三个整数Sj, Ej和Lj。

输出格式
输出一个整数, 所需要的总时间(包含和在你所在的牧场的奶牛的两次谈话时间)。
样例输入
5 7
10
10
20
6
30
1 2 5
2 3 5
2 4 12
3 4 17
2 5 15
3 5 6
样例输出
176
数据规模与约定
5 <= N <= 10000,N-1 <= P <= 100000,0 <= Lj <= 1000,1 <= Ci <= 1,000。

思路:最小生成树变形,kruskal()算法

//
//  main.cpp
//  算法训练 安慰奶牛
//
//  Created by nuu_tong on 2018/1/16.
//  Copyright © 2018年 apple. All rights reserved.
//  对于每条边,我们花费的时间是二倍的边权值加上两个端点的权值。
//  然后用这个花费作为边权,求最小生成树。
//  然后要选择权值最小的点作为根
//  将答案加上它就行了
#include <iostream>
#include <algorithm>
using namespace std;
const int Maxn=10010;
int father[Maxn],talkTime[Maxn];
int N,P;
struct edge
{
    int f,t,w;
    bool friend operator<(edge e1,edge e2)
    {
        return e1.w<e2.w;
    }
}E[100005];
int find(int x)
{
    if(father[x]==x)
        return x;
    else
        return father[x]=find(father[x]);
}
int Kruskal()
{
    int k,j;
    int n=1,totalTime=0;
    sort(E,E+P+1);
    for(int i=1;i<=P;i++)
    {
        k=find(E[i].f);
        j=find(E[i].t);
        if(k!=j)
        {
            father[j]=k;
            totalTime+=E[i].w;
            n++;
        }
        if(n==N)
            return totalTime;
    }
    return 0;
}
int main()
{
    int s,t,c;
    int sum;
    cin>>N>>P;
    for(int i=1;i<=N;i++)
    {
        father[i]=i;
        cin>>talkTime[i];
        //m=min(m,talkTime[i]);
    }
    for(int i=1;i<=P;i++)
    {
        cin>>s>>t>>c;
        E[i].f=s;
        E[i].t=t;
        E[i].w=(c<<1)+talkTime[s]+talkTime[t];
    }
    sum=Kruskal();
    sort(talkTime,talkTime+N+1);
    cout<<sum+talkTime[1]<<endl;
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值