LeetCode841钥匙和房间

题目描述

  有 n 个房间，房间按从 0 到 n - 1 编号。最初，除 0 号房间外的其余所有房间都被锁住。你的目标是进入所有的房间。然而，你不能在没有获得钥匙的时候进入锁住的房间。当你进入一个房间，你可能会在里面找到一套不同的钥匙，每把钥匙上都有对应的房间号，即表示钥匙可以打开的房间。你可以拿上所有钥匙去解锁其他房间。给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。如果能进入 所有 房间返回 true，否则返回 false。

解析

  用队列或者图的深度优先搜索。用队列会慢一点。
用队列：

public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        Queue<List<Integer>> queue = new ArrayDeque<>();
        queue.offer(rooms.get(0));
        boolean[] canOpen = new boolean[rooms.size()];
        canOpen[0] = true;
        int counter = 1;
        while(!queue.isEmpty()) {
            List<Integer> cur = queue.poll();
            for(int i : cur) {
                if(!canOpen[i]){
                    counter ++;
                    canOpen[i] = true;
                    queue.offer(rooms.get(i));
                }
            }
        }

        return counter == rooms.size();
    }

用DFS：

public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        boolean[] canOpen = new boolean[rooms.size()];
        DFS(rooms, 0, canOpen);
        for(boolean b : canOpen) {
            if(!b){
                return false;
            }
        }
        return true;
    }

    public void DFS(List<List<Integer>> G, int cur, boolean[] visited){
        visited[cur] = true;
        List<Integer> curList = G.get(cur);
        for(int i : curList){
            if(!visited[i]){
                DFS(G,i,visited);
            }
        }
    }