[Java] PAT甲级 1021 Deepest Root

代码可全部AC。在网上参考了一些思路，想明白了来总结学习一下。（刚写博客不久不会加图，就用文字说明了）

首先利用dfs看数据分为几部分。
如果是一个无环图（就是成树，这道题可能存在有环的情况，稍后说明），则存在一条最长路径AZ（两边缘可能包括多个点），在这条路径上有多个分支，任意取某一分支，假设该分支与最长路径交于M点，该分支最边上的点为N，那么一定存在这种情况NM<min(AM,MZ)。所以如果从该图的任意一点开始遍历，不断更新距离该点最远的点，那么最后肯定能得到最长路径两个边缘点集合的其中一个集合。同理从该集合中任选一点再开始遍历又得到另一端的点集合。两个集合合并为最终的结果集。
如果是不成树的情况（包括有环的情况）那么只需要知道数据共几部分即可。
这里需要注意最后的结果集要用Set容器，不能有重复数字！！！
代码如下:

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

public class Main {

	static int maxheight = 0;
	static int n = 0;
	static ArrayList<Integer>[] v;
	static boolean[] visit;
	static Set<Integer> s = new HashSet<>();
	static ArrayList<Integer> temp = new ArrayList<>();

	@SuppressWarnings("unchecked")
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
		n = Integer.parseInt(bf.readLine());
		v = new ArrayList[n + 1];
		visit = new boolean[n + 1];
		for (int i = 0; i < v.length; i++) {
			v[i] = new ArrayList<>();
		}

		int cnt = 0;
		int s1 = 0;
		for (int i = 0; i < n - 1; i++) {
			String[] bLine = bf.readLine().split(" ");
			int a = Integer.parseInt(bLine[0]);
			int b = Integer.parseInt(bLine[1]);

			v[a].add(b);
			v[b].add(a);
		}
		for (int i = 1; i <= n; i++) {
			if (visit[i] == false) {
				dfs(i, 1);
				if (i == 1) {
					if (temp.size() != 0)
						s1 = temp.get(0);
					for (int j = 0; j < temp.size(); j++)
						s.add(temp.get(j));
				}
				cnt++;
			}
		}
		if (cnt >= 2) {
			System.out.println("Error: " + cnt + " components");
		} else {
			temp.clear();
			maxheight = 0;
			visit = new boolean[n + 1];
			dfs(s1, 1);
			for (int i = 0; i < temp.size(); i++)
				s.add(temp.get(i));
			int[] arr = new int[s.size()];
			int i = 0;
			Iterator<Integer> it = s.iterator();
			while(it.hasNext()){
				arr[i++] = it.next();
			}
			Arrays.sort(arr);
			for (int b : arr)
				System.out.println(b);
		}
	}

	static void dfs(int node, int height) {
		if (height > maxheight) {
			temp.clear();
			temp.add(node);
			maxheight = height;
		} else if (height == maxheight) {
			temp.add(node);
		}
		visit[node] = true;
		for (int i = 0; i < v[node].size(); i++) {
			if (visit[v[node].get(i)] == false)
				dfs(v[node].get(i), height + 1);
		}
	}
}