BZOJ 1232: [Usaco2008Nov]安慰奶牛cheer Kruskal最小生成树

Description

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

• 第 1 行: 用空格隔开的两个整数N和P * 第 2..N+1 行: 第i+1行包含了一个整数: C_i * 第 N+2..N+P+1 行: 第 N+j+1 行包含用空格隔开的三个整数: S_j, E_j 和 L_j
  Output

第 1 行: 一个整数, 所需要的总时间(包含和在你所在的牧场的奶牛的两次谈话时间).

Sample Input
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

4 5 12

Sample Output
176

解法：

对于每条边，我们花费的时间是二倍的边权值加上两个端点的权值。

然后用这个花费作为边权，求最小生成树。

然后要选择权值最小的点作为根

将答案加上它就行了

//BZOJ 1232

#include <bits/stdc++.h>
using namespace std;
const int maxn = 10010;
int n, p, c[maxn], fa[maxn];
int tot, ans;
struct node{
    int u, v, w;
    node(){}
    node(int u, int v, int w) : u(u), v(v), w(w) {}
}E[maxn*10];
bool cmp(node a, node b){
    return a.w < b.w;
}
int find_set(int x){
    if(x == fa[x]) return x;
    else return fa[x] = find_set(fa[x]);
}
int main()
{
    scanf("%d%d", &n, &p);
    ans = 0x3f3f3f3f;
    for(int i=1; i<=n; i++){
        scanf("%d", &c[i]);
        ans = min(ans, c[i]);
        fa[i]=i;
    }
    for(int i=1; i<=p; i++){
        int u, v, w;
        scanf("%d%d%d", &u, &v, &w);
        w = w*2+c[u]+c[v];
        E[i].u=u,E[i].v=v,E[i].w=w;
    }
    sort(E+1, E+p+1, cmp);
    for(int i=1; i<=p; i++){
        int x = E[i].u, y = E[i].v;
        int fx = find_set(x), fy = find_set(y);
        if(fx != fy){
            fa[fx] = fy;
            tot++;
            ans+=E[i].w;
        }
        if(tot==n-1) break;
    }
    printf("%d\n", ans);
    return 0;
}