Problem Description 上次Gardon的迷宫城堡小希玩了很久(见Problem B),现在她也想设计一个迷宫让Gardon来走。但是她设计迷宫的思路不一样,首先她认为所有的通道都应该是双向连通的,就是说如果有一个通道连通了房间A和B,那么既可以通过它从房间A走到房间B,也可以通过它从房间B走到房间A,为了提高难度,小希希望任意两个房间有且仅有一条路径可以相通(除非走了回头路)。小希现在把她的设计图给你,让你帮忙判断她的设计图是否符合她的设计思路。比如下面的例子,前两个是符合条件的,但是最后一个却有两种方法从5到达8。
Input 输入包含多组数据,每组数据是一个以0 0结尾的整数对列表,表示了一条通道连接的两个房间的编号。房间的编号至少为1,且不超过100000。每两组数据之间有一个空行。
Output 对于输入的每一组数据,输出仅包括一行。如果该迷宫符合小希的思路,那么输出"Yes",否则输出"No"。
Sample Input 6 8 5 3 5 2 6 4 5 6 0 0 8 1 7 3 6 2 8 9 7 5 7 4 7 8 7 6 0 0 3 8 6 8 6 4 5 3 5 6 5 2 0 0 -1 -1
Sample Output Yes Yes No
|
import java.util.HashSet;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int x = sc.nextInt();
int y = sc.nextInt();
while (x != -1 && y != -1) {
if (x == 0 && y == 0){//为什么0 0要输出Yes
System.out.println("Yes");
x = sc.nextInt();
y = sc.nextInt();
continue;
}
HashSet<Integer> hashSet = new HashSet<>();
boolean b = false;
int[] arr = new int[100001];
for (int i = 1; i < arr.length; i++) {
arr[i] = i;
}
while (x != 0 && y != 0) {
hashSet.add(x);
hashSet.add(y);
int xF = find(arr, x);
int yF = find(arr, y);
if (x != y){//卡了我很久的一个bug
if (xF == yF) {
b = true;
} else {
arr[xF] = yF;
}
}
x = sc.nextInt();
y = sc.nextInt();
}
int sum = hashSet.size();
int reSum = 0;
for (int i = 1; i < arr.length; i++) {//计算有效节点个数
if (hashSet.contains(i) && arr[i] != i) {
reSum++;
}
}
reSum = sum - reSum;//计算根节点的个数
if (b || reSum > 1) {
System.out.println("No");
} else {
System.out.println("Yes");
}
x = sc.nextInt();
y = sc.nextInt();
}
sc.close();
}
private static int find(int[] arr, int x) {
int res = x;
while (res != arr[res]) {
res = arr[res];
}
while (x != res) {
int temp = arr[x];
arr[x] = res;
x = temp;
}
return res;
}
}