蓝桥杯 安慰奶牛

问题描述
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 6
10
10
20
6
30
1 2 5
2 3 5
2 4 12
3 4 17
2 5 15
3 5 6
样例输出
176

说明一下，这道题的样例输出是错误的，正确的是178，这不会影响到提交后的正误判断。

这道题很明显是一道涉及最小生成树的题目，说到最小生成树，我们都知道该算法需要通过判断图的边的权值来实现，如果用在这道题上，我们可以通过最小生成树求所有Lj的和，但这不足以解决问题，因为安慰奶牛的时间没有算上，如果我们先求所有Lj的和再加上安慰奶牛的时间就会变得很复杂。其实解决这道题是有技巧的，我们可以把Lj*2+边连接的两个安慰奶牛的时间作为边的权值，最后我们用最小生成树算法求出权值的和后在加上Ci中的最小值（在Ci最小的地方过夜耗时是最短的）。自己想一下，因为每次你到达第i个牧场的时候，你必须花去Ci的时间和奶牛交谈，即使你已经到过。所以即便多次从一个点出发，这点的时间也要算多次，因此，每走一条边，我们都要把边连接的两个点时间算上，这样就不用再单独算安慰奶牛的时间了。每条边都要走两次，一去一回，所以权值 = Li * 2 +第一个点的时间 +第二个点的时间。

代码如下：

#include <stdio.h>
#include <stdlib.h>

struct e
{
	int weight;//边的权值
	int u;//边连接的第一点
	int v;//边连接的第二点
}edge[10000];//构建图的边集

int father[10000] = {0};

void disjointSet_init(int n)
{
	int i;

	for (i = 0; i <= n; i++)
		father[i] = i;
}

int disjointSet_find(int element)
{
	if (father[element] == element)
		return element;
	else
	{
		father[element] = disjointSet_find(father[element]);
		return father[element];
	}
}

void  disjointSet_union(int a, int b)
{
	father[disjointSet_find(a)] = disjointSet_find(b);
}

int cmp(const void* a, const void* b)//qsort函数需要用到的函数，不懂qsort函数怎么使用可以自行去了解
{
	struct e* pa = (struct e*)a;
	struct e* pb = (struct e*)b;

	return pa->weight - pb->weight;
}

int kruskal(int N, int P)//使用最小生成树中的kruskal算法
{
	int i;
	int sum = 0;
	int count = 0;

	disjointSet_init(P);
	qsort(edge + 1, P, sizeof(struct e), cmp);//对边进行快速排序

	for (i = 1; i <= P; i++)
	{
		if (disjointSet_find(edge[i].u) != disjointSet_find(edge[i].v))
		{
			disjointSet_union(edge[i].u, edge[i].v);
			sum += edge[i].weight;//求权值和
			count ++;
		}

		if (count == N - 1)
			break;
	}

	return sum;
}

int main()
{
	int N, P, Ci, Sj, Ej, Lj;
	int i;
	int C[10000] = {0};
	int sum = 0;
	int min = 1000000000;

	scanf("%d%d", &N, &P);

	for (i = 1; i <= N; i++)
	{
		scanf("%d", &C[i]);

		if (min > C[i])
			min = C[i];//找出Ci中的最小值
	}
	
	for (i = 1; i <= P; i++)
	{
		scanf("%d%d%d", &edge[i].u, &edge[i].v, &edge[i].weight);
		edge[i].weight <<= 1;
		edge[i].weight += C[edge[i].u] + C[edge[i].v];//前面描述说的公式
	}

	sum = kruskal(N, P);

	sum += min;//权值和加上Ci最小值
	printf("%d\n", sum);

	return 0;
}