算法训练 安慰奶牛

问题描述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。--------------------------------------------------------------------------这题题意有点难理解，说的是要所有牧场连通，但是要使自己总花费时间最少去安慰完奶牛。自己看了网上大神写的代码，每条边的权重可以化为两个牧场安慰时间加上路途花费时间的两倍。因为树的回路经过每条边两次。我用的是kruskal算法。kruskal是将每条边按权重从小到大排序，然后从小到大依次取出每一条边，但是有个条件，这每一条边的两个端点至少有一个是之前取出的边没有的端点；不然将会形成回路。按照这个思路写成代码。自己却测评只得了10分，很多超时，尚未解决；等以后有空在想想该如何处理。import java.util.Scanner;

public class Main {
public static void main(String[] args) {
Scanner reader = new Scanner(System.in);
int N = reader.nextInt();
int P = reader.nextInt();
Node tree = new Node();
int[] cow = new int
;
Boolean[] finish = new Boolean
;
for (int i = 0; i < N; i++) {
tree = new Node();
cow[i] = reader.nextInt();
finish[i] = false;
}
Node node = tree;
for (int i = 0; i < P; i++) {
int top = reader.nextInt() - 1;
int down = reader.nextInt() - 1;
int quality = reader.nextInt();
node.data = quality*2+cow[top]+cow[down];
node.top = top;
node.down = down;
node.next = new Node();
node = node.next;
}

int min = cow[0];
int fromAction = 0;
// 寻找最短交流住所
for (int i = 0; i < N; i++)
if (min > cow[i]) {
min = cow[i];
fromAction = i;
}
int position = 0;

// 排序
Node sort = new Node();
node = tree;
Node node2 = sort;
while (node.data != Integer.MAX_VALUE) {
if (node2.next == null) {
node2.data = node.data;
node2.top = node.top;
node2.down = node.down;
node2.next = new Node();
node2 = sort;
node = node.next;
} else if (node.data>=node2.data&&node.data <=node2.next.data) {

Node newNode = new Node();
newNode.data = node.data;
newNode.down = node.down;
newNode.top = node.top;
newNode.next = node2.next;
node2.nex
4000
t = newNode;
node2 = sort;
node = node.next;
} else
node2 = node2.next;

}
//生成最小生成树
tree=new Node();
node=tree;
while(node2.data!=Integer.MAX_VALUE)
{
if(!(finish[node2.down]&&finish[node2.top]))
{
finish[node2.down]=true;
finish[node2.top]=true;
node.data=node2.data;
node.down=node2.down;
node.top=node2.top;
node.next=new Node();
node=node.next;
node2=node2.next;

}
else
node2=node2.next;

}
int result=0;
node2=tree;
int number[]=new int
;
while (node2.data != Integer.MAX_VALUE) {
result+=node2.data;
number[node2.top]++;
number[node2.down]++;
node2 = node2.next;
}
System.out.println(result+cow[fromAction]);
}
}
class Node {
Node next=null;
int top;
int down;
int data=Integer.MAX_VALUE;
}