题目描述
有 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。
分析
这道题主要是考察深度搜索
递归出口,维护一个计算器,当计数器的值达到n的时候就算找到了
使用标记数组,标记当前的房间是够已经走过
方法一
class Solution {
public:
//这道题主要是考察深度搜索
//递归出口,维护一个计算器,当计数器的值达到n的时候就算找到了
//使用标记数组,标记当前的房间是够已经走过
bool dfs(vector<vector<int>>& rooms, int index, int cnt, vector<int>& vis){
if(vis[index]) return cnt==rooms.size();
vis[index] = 1;
if(cnt==rooms.size()){
//cout<<cnt;
return true;
}
for(int i=0;i<rooms[index].size();i++){
bool res = dfs(rooms, rooms[index][i], cnt+1, vis);
if(res) return res;
}
return false;
}
bool canVisitAllRooms(vector<vector<int>>& rooms) {
vector<int> vis(rooms.size(), 0);
//vis[0] = 1;
bool ans = dfs(rooms, 0, 0, vis);
for(auto it:vis){
if(it==0) return false;
}
return true;
}
};
方法二
class Solution {
public:
int num = 0;//总的计数器
vector<int> vis;
//index表示房间的索引号
void dfs(vector<vector<int>>& rooms, int index){
vis[index] = 1;
num++;
for(int i=0;i<rooms[index].size();i++){
if(!vis[rooms[index][i]]){
dfs(rooms, rooms[index][i]);
}
}
}
bool canVisitAllRooms(vector<vector<int>>& rooms) {
vis.resize(rooms.size());
dfs(rooms, 0);
return num==rooms.size();
}
};