有 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。
示例 1:
輸入: [[1],[2],[3],[]]
輸出: true
解釋:
我們從 0 號房間開始,拿到鑰匙 1。
之后我們去 1 號房間,拿到鑰匙 2。
然后我們去 2 號房間,拿到鑰匙 3。
最后我們去了 3 號房間。
由于我們能夠進入每個房間,我們返回 true。
示例 2:
輸入:[[1,3],[3,0,1],[2],[0]]
輸出:false
解釋:我們不能進入 2 號房間。
提示:
1 <= rooms.length <= 1000
0 <= rooms[i].length <= 1000
所有房間中的鑰匙數量總計不超過 3000。
解題思路:與332相似,都為圖的Dfs遍歷
int count=0;
bool canVisitAllRooms(vector<vector<int>>& rooms) {
vector<int> flag(rooms.size());
Dfs(0,rooms,flag);
if(count==rooms.size())
return true;
return false;
}
void Dfs(int pos,vector<vector<int>>& rooms,vector<int>& flag)
{
if(flag[pos]==0)
{
flag[pos]=1;
for(auto i:rooms[pos])
{
Dfs(i,rooms,flag);
}
count++;
}
}