leetcode 841. 钥匙和房间

题目描述

有 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();
    }
};