題目
有 N 個房間,開始時你位于 0 號房間。每個房間有不同的號碼:0,1,2,...,N-1,并且房間里可能有一些鑰匙能使你進(jìn)入下一個房間。
在形式上,對于每個房間 i 都有一個鑰匙列表 rooms[i],每個鑰匙 rooms[i][j] 由 [0,1,...,N-1] 中的一個整數(shù)表示,其中 N = rooms.length。 鑰匙 rooms[i][j] = v 可以打開編號為 v 的房間。
最初,除 0 號房間外的其余所有房間都被鎖住。
你可以自由地在房間之間來回走動。
如果能進(jìn)入每個房間返回 true,否則返回 false。
示例 1:
輸入: [[1],[2],[3],[]]
輸出: true
解釋:
我們從 0 號房間開始,拿到鑰匙 1。
之后我們?nèi)?1 號房間,拿到鑰匙 2。
然后我們?nèi)?2 號房間,拿到鑰匙 3。
最后我們?nèi)チ?3 號房間。
由于我們能夠進(jìn)入每個房間,我們返回 true。
示例 2:
輸入:[[1,3],[3,0,1],[2],[0]]
輸出:false
解釋:我們不能進(jìn)入 2 號房間。
提示:
1. 1 <= rooms.length <= 1000
2. 0 <= rooms[i].length <= 1000
3. 所有房間中的鑰匙數(shù)量總計不超過 3000。
解答
class Solution {
boolean[] vis;
int cnt;//遍歷節(jié)點數(shù)
public boolean canVisitAllRooms(List<List<Integer>> rooms) {
int n = rooms.size();
cnt = 0;
vis = new boolean[n];
dfs(rooms, 0);
return cnt == n;
}
public void dfs(List<List<Integer>> rooms, int room) {
vis[room] = true;
cnt++;
for (int ro : rooms.get(room)) {
if (!vis[ro]) {
dfs(rooms, ro);
}
}
}
}