算法训练 安慰奶牛 (Kruskal-最小生成树)

问题描述
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。

题目给的数据貌似有问题,照着网上看到的其他数据做了测试。
每个牧场间只有一条通路,还要安慰的时间最小,其实就是一个最小生成树的题。安慰时间Ci是点的权值,Li是边的权值。题目还要选定一个点作为往返点,自己拿书上一个连通图做了测试,任意选择节点遍历,发现只有起始点是被经过次数最多的,那么要满足题目要求时间最小,即选择权值最小的节点作为起点,边权×2加上点权作为Kruskal的权重,最后加上一次起点权值,就是题目的答案。

#include<iostream>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define MAX 100002 //最大边数
typedef struct Edge{
    int S,E,L;
}Edge;

Edge e[MAX];

int parent[MAX]; //指向根节点的节点数组
//约定单个的节点数组值为-1,数组值的绝对值代表节点个数
int C[MAX];//每个点的权值
int sum=0;
int N,P;

int cmp(const void *a,const void *b)//调用qsort()快排函数的函数指针
{
    Edge *c=(Edge *)a;
    Edge *d=(Edge *)b;
    return c->L - d->L;
}

int Find(int x){//查集,即返回该节点的根节点
    int s;
    for(s=x;parent[s]>=0;s=parent[s]);//这一句就是把x的根节点给了s
    while(s!=x)//优化策略--压缩路径
    {
     int tmp=parent[x];
     parent[x]=s;
     x=tmp;
    }
    return s;
}

void Union(int S,int E){//并集
    int r1=Find(S),r2=Find(E);//r1,r2分别为S和E的根节点
    int tmp=parent[r1]+parent[r2];//tmp为负数,所以下面的比较要反着看
    if(parent[r1]>parent[r2])//加权合并--如果r2小于r1
    //那么将两个集合中元素个数少的根节点作为元素个数多的根节点的子节点
    {
     parent[r1]=r2;
     parent[r2]=tmp;
    }
    else//反之
    {
     parent[r2]=r1;
     parent[r1]=tmp;
    }
}

int Kruskal(){
    int S,E;
    int sumweight=0,count=0;
    for(int i=0;i<N;i++)//初始化parent数组
     parent[i]=-1;
    qsort(e,P,sizeof(e[0]),cmp);//将所有边的权值按从小到大排序
    //这里边的权值是L=Li*2+Cs+Ce
    for(int j=0;j<P;j++)
    {
     S=e[j].S;
     E=e[j].E;
     if(Find(S)!=Find(E))//如果两个节点的根节点不相同,说明没有生成回路,那么合并这两个集合
    {
     sumweight+=e[j].L;
     Union(S,E);
     count++;
     if(count >=N-1) {break;}
    }
    }
        return sumweight;
}

int main()
{
    int i,min=MAX;
    int S,E,L;
    cin>>N>>P;
    for(i=0;i<N;i++)
    {
     cin>>C[i];
     if(C[i]<min) min=C[i];//找出权值最小的那个点,就是睡觉的地方,最后要加上这一次睡觉时间
    }
    for(i=0;i<P;i++)
    {
     cin>>S>>E>>L;
     e[i].S=S-1;
     e[i].E=E-1;
     e[i].L=L*2+C[S-1]+C[E-1];
    }
    cout<<min+Kruskal()<<endl;
    return 0;
}




  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值