Algo-6-安慰奶牛

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

问题分析
一个求最小生成树的问题。特殊点在于,由于每晚都要在同一个牧场过夜,猎人出去了还要原路返回(原路返回是最优的),
所以每条边都要走两次。只要经过一条边,势必要经过它两端的结点。所以每条边的权值可以如下设置:原来的权值*2+两端结点的权值
由于已经是最小生成树了,无论猎人从那个结点出发都要走过所有的边,那就让猎人从权值最小的结点出发,时间是最省的
所以所有边的权值之和,再加上最小的结点权值即为答案。
最小的结点只需加一次。相当于加的是晚上睡觉的那一次,出发时的那一次已经算在连接该结点的边的权值里了。 
源代码采用Kruskal(克鲁斯卡尔)算法求解最小生成树。

#include <iostream>
#include <algorithm>

using namespace std;

const int MAXN = 10000+10;
const int MAXP = 100000+10;
const int INF = 0xFFFFFFF;

struct edge{
	int s;int e;int value;
	edge(){};
	edge(int s,int e,int v){
		this->s = s;
		this->e = e;
		this->value = v;
	}
}edges[MAXP];

int n,p; 
int ci[MAXN];
int parent[MAXN];

void UFset(){
	for(int i = 0;i<=n;i++)
		parent[i] = -1;
}

int Find(int x){
	int s;
	for(s = x;parent[s] >= 0;s=parent[s])
		;
	while(s != x){
		int tmp = parent[x];
		parent[x] = s;
		x = tmp;
	}
	return s;
}

void Union(int R1,int R2){
	int r1 = Find(R1);
	int r2 = Find(R2);
	int tmp = parent[r1]+parent[r2];
	if(parent[r1] > parent[r2]){
		parent[r1] = r2;
		parent[r2] = tmp;
	}else{
		parent[r2] = r1;
		parent[r1] = tmp;
	}
}
 
//实现从小到大排序函数 
int cmp(const void *a,const void *b){
	edge e1 = *(const edge *)a;
	edge e2 = *(const edge *)b;
	return e1.value - e2.value;
}

int Kruskal(){
	int sumweight = 0;
	int num = 0;
	int s,e;
	UFset();
	for(int i = 0;i<p;i++){
		s = edges[i].s;
		e = edges[i].e;
		if(Find(s) != Find(e)){
			num++;
			sumweight += edges[i].value;
			Union(s,e);
		}
		if(num > n-1)
			break;
	}
	
	return sumweight;
	
}

int main(){
	int s,e,l;
	cin >> n >> p;
	int min = INF;   //用于保存结点的最小权值,猎人最后就从这个结点出发 
	for(int i = 1;i<=n;i++){
		cin >> ci[i];
		if(ci[i] < min)
			min = ci[i];
	}
	for(int i = 0;i<p;i++){
		cin >> s >> e >> l;
		int newv = ci[s] + ci[e] + l*2; 
		edges[i] = edge(s,e,newv);
	}
	
	qsort(edges,p,sizeof(edges[0]),cmp);
	int res = Kruskal();
	cout << res + min << endl;
	
	return 0;
} 

求解最小生成树的算法

Kruskal算法
1.构造一个只含n个结点,而边集为空的子图
2.对边集按权值从小到大排序。每次选择权值最小的边,若边的两端点属于不同的子树,则将该边选入子图(该边的两端点则属于同一棵树);否则,舍弃该边。继续循环
3.若子图中边的数目等于n-1,停止循环。此时得到的子图即为最小生成树

Prim算法
1.从点击V中任选一结点作为起始结点,加入已选结点集合visit;初始化已选边集edge
2.每次都从剩下点集中选择距离集合visit最短的结点,加入visit,将距离最短的边加入edge(所谓距离集合visit最短的结点,指与集合中的点相连的所有非visit结点中权值最小的那个)
3.当所有结点都属于visit时,停止算法。edge中保存的即为最小生成树的边

Kruskal算法的复杂度为O(eloge),Prim算法的复杂度为O(n^2).当边的数目较多时用Prim算法更合适。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值