输入
每个测试点(输入文件)有且仅有一组测试数据。
每组测试数据的第1行为一个整数N,意义如前文所述。
每组测试数据的第2~N+1行,每行分别描述一对父子关系,其中第i+1行为两个由大小写字母组成的字符串Father_i, Son_i,分别表示父亲的名字和儿子的名字。
每组测试数据的第N+2行为一个整数M,表示小Hi总共询问的次数。
每组测试数据的第N+3~N+M+2行,每行分别描述一个询问,其中第N+i+2行为两个由大小写字母组成的字符串Name1_i, Name2_i,分别表示小Hi询问中的两个名字。
对于100%的数据,满足N<=10^2,M<=10^2, 且数据中所有涉及的人物中不存在两个名字相同的人(即姓名唯一的确定了一个人)。
输出
对于每组测试数据,对于每个小Hi的询问,输出一行,表示查询的结果:如果根据已知信息,可以判定询问中的两个人存在共同的祖先,则输出他们的所有共同祖先中辈分最低的一个人的名字,否则输出-1。
思路:
先用其中一个来回溯到根,记录路径上的节点为访问过,再用另一个来回溯到根,期间只要第一次碰到已经访问过的节点,就是答案(不可以同时回溯)
public class Main {
static Map<String, String> sonFather; //son to father
static Set<String> visited;
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
int n = Integer.parseInt(in.nextLine());
sonFather = new HashMap<String, String>();
visited = new HashSet<String>();
for (int i = 0; i < n; i++) {
String[] s = in.nextLine().split(" ");
sonFather.put(s[1], s[0]);
}
int m = Integer.parseInt(in.nextLine());
for (int i = 0; i < m; i++) {
String[] s = in.nextLine().split(" ");
quest(s[0],s[1]);
visited = new HashSet<String>();
}
in.close();
}
public static void quest(String s1, String s2) {
if (s1.equals(s2)) {
System.out.println(s1);
return;
}
if (!sonFather.containsKey(s1) && !sonFather.containsKey(s2)) {
System.out.println(-1);
return;
}
while (sonFather.containsKey(s1)) {
visited.add(s1);
s1 = sonFather.get(s1);
}
visited.add(s1);
while (sonFather.containsKey(s2) && !visited.contains(s2)) {
s2 = sonFather.get(s2);
}
if (visited.contains(s2)) {
System.out.println(s2);
return;
}
System.out.println(-1);
}
}