LeetCode 841. Keys and Rooms 钥匙和房间（Java）

最新推荐文章于 2023-09-25 14:50:22 发布

volador_r

最新推荐文章于 2023-09-25 14:50:22 发布

阅读量334

点赞数

分类专栏： LeetCode 文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/qq_41878358/article/details/125563880

版权

LeetCode 专栏收录该内容

147 篇文章 2 订阅

订阅专栏

题目：

There are n rooms labeled from 0 to n - 1 and all the rooms are locked except for room 0. Your goal is to visit all the rooms. However, you cannot enter a locked room without having its key.

When you visit a room, you may find a set of distinct keys in it. Each key has a number on it, denoting which room it unlocks, and you can take all of them with you to unlock the other rooms.

Given an array rooms where rooms[i] is the set of keys that you can obtain if you visited room i, return true if you can visit all the rooms, or false otherwise.

Example 1:
Input: rooms = [[1],[2],[3],[]]
Output: true
Explanation:
We visit room 0 and pick up key 1.
We then visit room 1 and pick up key 2.
We then visit room 2 and pick up key 3.
We then visit room 3.
Since we were able to visit every room, we return true.

Example 2:
Input: rooms = [[1,3],[3,0,1],[2],[0]]
Output: false
Explanation: We can not enter room number 2 since the only key that unlocks it is in that room.

Constraints:

n == rooms.length
2 <= n <= 1000
0 <= rooms[i].length <= 1000
1 <= sum(rooms[i].length) <= 3000
0 <= rooms[i][j] < n
All the values of rooms[i] are unique.

解答：

题目明显可以采用递归的思路。获取当前第k个房间的钥匙🔑，通过递归，依次进入第k个房间所能拿到的钥匙对应的房间… …若当前房间已经能被访问，则退出递归。通过 boolean flag[]标记哪些房子可以被访问。

class Solution {
    public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        if(rooms.size() == 0) {
            return true;
        }
        int n = rooms.size(); //房间的数量
        boolean flag[] = new boolean[n];    //标志哪个房间可以被访问
        visitRoom(rooms, flag, 0);
        for(int r = 1; r < n; r++) {
            if(!flag[r]) {
                return false;
            }
        }
        return true;
    }
    public void visitRoom(List<List<Integer>> rooms, boolean flag[], int k) {
        if(flag[k]) {
            return;
        }
        flag[k] = true;
        List<Integer> getKey = rooms.get(k);
        for(int i = 0; i < getKey.size(); i++) {
            visitRoom(rooms, flag, getKey.get(i));
        }
    }
}

上述递归的方法，其实也就是我们常说的DFS深度优先搜索。
而递归其实就是隐式栈，所以我们也可以用显式栈来实现。栈的先进后出其实是实现回退上一步的方式。不管是显示栈调用还是隐式的方法递归都可以。
LeetCode上给出的Depth-First Search的Solution，其实就是用到了栈Stack。当访问当前房间时，将能够获取到的房间钥匙🔑入栈，并表示这些🔑对应的房间均可被访问。随后依次出栈继续搜索，直到栈为空，所有可被访问的房间均已被访问。

class Solution {
    public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        int n = rooms.size(); //房间的数量
        boolean flag[] = new boolean[n];    //标志哪个房间可以被访问
        flag[0] = true;
        Stack<Integer> stack = new Stack(); //用于记录当前可进入的房间号，依次入栈并出栈，实现递归
        stack.push(0);
        
        while(!stack.isEmpty()) {
            List<Integer> keyList = rooms.get(stack.pop());
            for(int key: keyList) {
                if(!flag[key]) {
                    flag[key] = true;
                    stack.push(key);
                }
            }
        }
        for(int r = 1; r < n; r++) {
            if(!flag[r]) {
                return false;
            }
        }
        return true;
    }
}

上述两种方法均采用了DFS，也可以采用BFS，借助队列实现：

class Solution {
    public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        int n = rooms.size(); //房间的数量
        boolean flag[] = new boolean[n];    //标志哪个房间可以被访问
        flag[0] = true;
        Queue<Integer> queue = new LinkedList<Integer>();
        queue.offer(0);
        while(!queue.isEmpty()) {
            List<Integer> keyList = rooms.get(queue.poll());
            for(int key: keyList) {
                if(!flag[key]) {
                    flag[key] = true;
                    queue.offer(key);
                }
            }
        }
        for(int r = 1; r < n; r++) {
            if(!flag[r]) {
                return false;
            }
        }
        return true;
    }
}