时间:20180906
题目描述 :
有 N 个房间,开始时你位于 0 号房间。每个房间有不同的号码:0,1,2,...,N-1,并且房间里可能有一些钥匙能使你进入下一个房间。
在形式上,对于每个房间 i 都有一个钥匙列表 rooms[i],每个钥匙 rooms[i][j] 由 [0,1,...,N-1] 中的一个整数表示, 其中 N = rooms.length。 钥匙 rooms[i][j] = v 可以打开编号为 v 的房间。
最初,除 0 号房间外的其余所有房间都被锁住。你可以自由地在房间之间来回走动。如果能进入每个房间返回 true,否则返回 false。要求先写出解题思路,再编写代码。
通过广度优先搜索(BFS)的方式进行探索。将可访问的0房间加入队列之中,访问队列中的一个元素,遍历他可以到达的房间,如果该房间还不能被访问,那么设置为访问,否则跳过。
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Stack;
public class KeysAndRooms {
public static void main(String[] args) {
//[[1],[2],[3],[]]
List<List<Integer>> rooms = new ArrayList();
List<Integer> keys0 = new ArrayList<>();
keys0.add(1);
List<Integer> keys1 = new ArrayList<>();
keys1.add(2);
List<Integer> keys2 = new ArrayList<>();
keys2.add(3);
List<Integer> keys3 = new ArrayList<>();
rooms.add(keys0);
rooms.add(keys1);
rooms.add(keys2);
rooms.add(keys3);
System.out.println(canVisitAllRooms(rooms));
}
private static boolean canVisitAllRooms(List<List<Integer>> rooms) {
//用栈存储下一步要参观的房间
Stack<Integer> dfs = new Stack<>();
dfs.add(0);
//用hashset存储已经参观过的房间
HashSet<Integer> seen = new HashSet<Integer>();
seen.add(0);
while (!dfs.isEmpty()) {
int i = dfs.pop();
for (int j : rooms.get(i))
if (!seen.contains(j)) {
dfs.add(j);
seen.add(j);
if (rooms.size() == seen.size()) return true;
}
}
return rooms.size() == seen.size();
}
}