算法训练 安慰奶牛

问题描述

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[N];
		Boolean[] finish = new Boolean[N];
		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.next = 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[N];
		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;
}