算法刷题-钥匙与房间

• 钥匙和房间

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

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

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

  • BFS

    /**
     * @param {number[][]} rooms
     * @return {boolean}
     */
    var canVisitAllRooms = function(rooms) {
    
        //bfs
    
        let visited = new Set() //定义已遍历过节点的集合
        let stack = [] //定义即将遍历节点的栈
        visited.add(0) //默认0是已遍历的节点 
        stack.push(0) //默认0是即将访问是节点
    
        while(stack.length){
        let nextRooms = rooms[stack.shift()] //下次访问的房间
        for(let i = 0;i<nextRooms.length;i++){
            if(!visited.has(nextRooms[i])){
                //已经遍历节点中没有，就向其添加
                visited.add((nextRooms[i]))
                stack.push(nextRooms[i])
            }
        }
       }
    
       return rooms.length == visited.size       
    };
    

  • DFS

    /**
     * @param {number[][]} rooms
     * @return {boolean}
     */
    var canVisitAllRooms = function(rooms) {
        //bfs
        let visited = new Set() //定义已遍历过节点的集合
        function dfs(current){
            visited.add(current)
            let nextRooms = rooms[current]//接下来要访问的房间
            for(let i =0;i<nextRooms.length;i++){
                if(!visited.has(nextRooms[i])){
                    dfs(nextRooms[i])
                }
            }
        }
       dfs(0)
       return rooms.length == visited.size       
    };
    

力扣参考