掘金 后端 ( ) • 2024-03-27 19:11

题目

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

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

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

示例 1:

输入: rooms = [[1],[2],[3],[]]
输出: true
解释:
我们从 0 号房间开始,拿到钥匙 1。
之后我们去 1 号房间,拿到钥匙 2。
然后我们去 2 号房间,拿到钥匙 3。
最后我们去了 3 号房间。
由于我们能够进入每个房间,我们返回 true。

示例 2:

输入: rooms = [[1,3],[3,0,1],[2],[0]]
输出: false
解释: 我们不能进入 2 号房间。

解法

我的解法不算最优,但直接用了dfs而且我觉得挺易懂的,这道题主要是考察图的遍历和dfs和bfs,因为图的遍历就是需要遍历整个图,找出从起始节点开始是否能够访问到所有节点。

class Solution {
    int N = 10010, M = N * 2 + 10, idx = 0, count = 1;
    boolean[] st = new boolean[N];
    List<List<Integer>> rooms;
    public boolean canVisitAllRooms(List<List<Integer>> room) {
        // Arrays.fill(h, -1);
        rooms = room;
        int n = rooms.size();
        dfs(0);
        if(count == n) {
            return true;
        }
        return false;
    }
    public void dfs(int nowx) {
        st[nowx] = true;
        List<Integer> nowList = rooms.get(nowx);
        System.out.println(nowx);
        for(Integer i : nowList) {
            if(st[i])    continue;
            dfs(i);
            count ++;
        }
    }
}