Cheering up the Cow
题目大意
约翰有N个牧场,编号依次为1到N。每个牧场里住着一头奶牛。连接这些牧场的有P条道
路,每条道路都是双向的。第j条道路连接的是牧场Sj和Ej,通行需要Lj的时间。两牧场之
间最多只有一条道路。约翰打算在保持各牧场连通的情况下去掉尽量多的道路。
约翰知道,在道路被强拆后,奶牛会非常伤心,所以他计划拆除道路之后就去忽悠她们。
约翰可以选择从任意一个牧场出发开始他维稳工作。当他走访完所有的奶牛之后,还要回
到他的出发地。每次路过牧场i的时候,他必须花Ci的时间和奶牛交谈,即使之前已经做过
工作了,也要留下来再谈一次。注意约翰在出发和回去的时候,都要和出发地的奶牛谈一
次话。请你计算一下,约翰要拆除哪些道路,才能让忽悠奶牛的时间变得最少?
分析
保持各牧场连通的情况下去掉尽量多的道路,提示了我们需要求最小生成树。而接下来的问题就是,求得最小的忽悠奶牛时间,对于一颗最小生成树,劝奶牛的路线如图,我们可以很明显的得出
- 每个牛的劝说次数等于其参与成边的边数及其度数,每条路都会经过两次
- 对于起点来说需要额外加上从其开始劝说的时间
- 这样不难得出,要让忽悠奶牛的时间变得最少,只需起点交谈时间最小
对于边权,我们处理为 两端点交谈时间之和 + 路途时间 * 2(根据其度数特点) 做最小生成树得到距离加上所选起始点的交谈时间即可
code
#include<bits/stdc++.h>
using namespace std;
const int N = 1e4 + 10, M = 2 * N;
typedef long long ll;
int n, m, k, t;
int arr[N], p[N];
struct Edge
{
int a, b, w;
bool operator < (const Edge & t) const
{
return w < t.w;
}
}e[M];
int findx(int x)
{
if(p[x] != x) p[x] = findx(p[x]);
return p[x];
}
int kruskal()
{
sort(e, e + m);
int res = 0;
for(int i = 0; i < m; i ++)
{
int a = findx(e[i].a), b = findx(e[i].b), w = e[i].w;
if(a != b)
{
p[a] = b;
res += w;
}
}
return res;
}
int main()
{
cin >> n >> m;
for(int i = 1; i <= n; i ++) p[i] = i;
int minn = 0x3f3f3f3f;
for(int i = 1; i <= n; i ++)
{
cin >> arr[i];
minn= min(arr[i], minn);
}
for(int i = 0; i < m; i ++)
{
int u, v, w;
cin >> u >> v >> w;
e[i] = {u, v, 2 * w + arr[u] + arr[v]};
}
int ans = kruskal();
cout << ans + minn << "\n";
return 0;
}