首页 > 学院 > 开发设计 > 正文

算法训练 安慰奶牛

2019-11-14 10:09:23
字体:
来源:转载
供稿:网友

问题描述 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。

package 安慰奶牛;import java.util.Arrays;import java.util.Comparator;import java.util.Scanner;class Graph implements Comparator<Graph>{ int from; int to; int weight; public int compare(Graph o1, Graph o2) { // TODO Auto-generated method stub return o1.weight-o2.weight; }}public class Main { static int[] PRe; static int[] time; static int N; static int P; static int min_time; static Graph[] graph; static int inf=1<<30; public static int Find(int x){ if ( x == pre[x]){ return x; }else{ int k = Find(pre[x]); pre[x] = k; return k; } } public static void main(String[] args) { // TODO Auto-generated method stub Scanner in = new Scanner(System.in); N = in.nextInt(); //牧场个数 P = in.nextInt(); //道路数量 time = new int[N+1]; //各个牧场安慰时间 pre = new int[N+1]; //前驱数组 min_time = inf; //最短时间 for ( int i = 1 ; i <= N ; i++){ pre[i] = i; time[i] = in.nextInt(); min_time = Math.min(min_time, time[i]); } graph = new Graph[P]; for ( int i = 0 ; i < P ; i++){ if (graph[i] == null) graph[i] = new Graph(); graph[i].from = in.nextInt(); graph[i].to = in.nextInt(); graph[i].weight = in.nextInt(); graph[i].weight = graph[i].weight * 2 + time[graph[i].from] + time[graph[i].to]; } Arrays.sort(graph,new Graph()); int sum = 0; for ( int i = 0 ; i < P ; i++){ int f1 = Find(graph[i].from); int f2 = Find(graph[i].to); if ( f1 != f2){ sum += graph[i].weight; pre[f1] = f2; } } System.out.println(sum+min_time); in.close(); }}

这里写图片描述


发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表