首页 > 代码库 > 算法训练 安慰奶牛   最小生成树

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

问题描述

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。

数据有问题 看代码里的
 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 100000000710 using namespace std;11 int n,p;12 struct node13 {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     else24         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 669 1070 1071 2072 673 3074 1 2 575 2 3 576 2 4 1277 3 4 1778 2 5 1579 3 5 680 */

 

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