算法训练 安慰奶牛   最小生成树

Posted 2020-09-07 半根毛线code

问题描述

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

输入格式

第1行包含两个整数N和P。

接下来N行，每行包含一个整数C_i。

接下来P行，每行包含三个整数S_j, E_j和L_j。

输出格式

输出一个整数, 所需要的总时间(包含和在你所在的牧场的奶牛的两次谈话时间)。

样例输入

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 <= L_j <= 1000，1 <= C_i <= 1,000。

数据有问题 看代码里的

 1 #include<bits/stdc++.h>
 2 #include<iostream>
 3 #include<cstring>
 4 #include<cstdio>
 5 #include<algorithm>
 6 #include<vector>
 7 #define ll __int64
 8 #define PI acos(-1.0)
 9 #define mod 1000000007
10 using namespace std;
11 int n,p;
12 struct node
13 {
14     int w;
15     int l,r;
16 }N[1000006];
17 int fa[100005];
18 int s[100005];
19 int find(int root)
20 {
21     if(fa[root]!=root)
22         return fa[root]=find(fa[root]);
23     else
24         return root;
25 }
26 void unio(int a,int b)
27 {
28     int aa=find(a);
29     int bb=find(b);
30     if(aa!=bb)
31     {
32         fa[aa]=bb;
33     }
34 }
35 bool cmp(struct node aa,struct node bb)
36 {
37     return aa.w<bb.w;
38 }
39 int main()
40 {
41     scanf("%d %d",&n,&p);
42     int minx=10000000;
43     for(int i=1;i<=n;i++)
44     {
45         scanf("%d",&s[i]);
46         minx=min(minx,s[i]);
47         fa[i]=i;
48     }
49     for(int i=1;i<=p;i++)
50     {
51         scanf("%d %d %d",&N[i].l,&N[i].r,&N[i].w);
52         N[i].w=2*N[i].w+s[N[i].l]+s[N[i].r];
53     }
54     sort(N+1,N+1+p,cmp);
55     int ans=0;
56     for(int i=1;i<=p;i++)
57     {
58         if(find(N[i].l)!=find(N[i].r))
59         {
60           ans=ans+N[i].w;
61           unio(N[i].l,N[i].r);
62         }
63     }
64     printf("%d\n",ans+minx);
65     return 0;
66 }
67 /*
68 5 6
69 10
70 10
71 20
72 6
73 30
74 1 2 5
75 2 3 5
76 2 4 12
77 3 4 17
78 2 5 15
79 3 5 6
80 */