leetcode841. 钥匙和房间(图的遍历DFS BFS)

• https://leetcode.cn/problems/keys-and-rooms

• 有 n 个房间，房间按从 0 到 n - 1 编号。最初，除 0 号房间外的其余所有房间都被锁住。你的目标是进入所有的房间。然而，你不能在没有获得钥匙的时候进入锁住的房间。

• 当你进入一个房间，你可能会在里面找到一套不同的钥匙，每把钥匙上都有对应的房间号，即表示钥匙可以打开的房间。你可以拿上所有钥匙去解锁其他房间。

• 给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。如果能进入 所有 房间返回 true，否则返回 false。

• 本题实际上是一个有向图的遍历问题

DFS

#include <iostream>
#include<memory>
#include<vector>
using namespace std;
class Soluton{
    public:
        void dfs(int root,vector<int>& visit,vector<vector<int>>& rooms){//不加&不会修改 vector<int>&
            if(visit[root]==1){
                return;
            }else{
                visit[root]=1;
            }
            for(int v:rooms[root]){
                dfs(v,visit,rooms);
            }
            
        }
        
        bool canVisitAllRooms(std::vector<vector<int>>& rooms){
            int n = rooms.size();//房间的个数
            vector<int> visit(n,0);
            visit[0] = 1;
            for(int v:rooms[0]){
                dfs(v,visit,rooms);
            }
            for(int invisit:visit){
                if(invisit == 0){
                    return false;
                }
            }
            return true;
        }
};

int main()
{
    cout<<"Hello World";
    //vector<vector<int>> rooms = {{1},{2},{3},{}};
    vector<vector<int>> rooms = {{1,3},{3,0,1},{2},{0}};
    
    unique_ptr<Soluton> mysolo = unique_ptr<Soluton>(new Soluton());
    bool res = mysolo->canVisitAllRooms(rooms);
    cout<< res;
    
    return 0;
}

BFS

#include <iostream>
#include<memory>
#include<vector>
#include<queue>
using namespace std;
class Soluton{
    public:
        std::queue<int> q;
        bool canVisitAllRooms(std::vector<vector<int>>& rooms){
            int n = rooms.size();//房间的个数
            vector<int> visit(n,0);
            visit[0] = 1;
            
            for(int v:rooms[0]){
                q.push(v);
            }
            while(!q.empty()){
                int cur = q.front(); q.pop();
                visit[cur] =1;

                for(int v:rooms[cur]){
                    if(visit[v] == 0){//防止死循环
                        q.push(v);
                    }
                }
            }
            
            
            for(int invisit:visit){
                if(invisit == 0){
                    return false;
                }
            }
            return true;
        }
};

int main()
{
    cout<<"Hello World";
    vector<vector<int>> rooms = {{1},{2},{3},{}};
    //vector<vector<int>> rooms = {{1,3},{3,0,1},{2},{0}};
    
    unique_ptr<Soluton> mysolo = unique_ptr<Soluton>(new Soluton());
    bool res = mysolo->canVisitAllRooms(rooms);
    cout<< res;
    
    return 0;
}

• 第二个测试用例