图论|841钥匙和房间

没脑袋的喵2023-11-30 14:32

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

java 复制代码

class Solution {
    public boolean canVisitAllRooms(List<List<Integer>> rooms) {
        //可以遍历所有房间指 从起点开始所有的点都可达
        //可以用栈解决
        //栈存放钥匙
        //hashset存放已被解锁的房间
        Stack<Integer> key=new Stack<Integer>();
        HashSet<Integer> room=new HashSet<Integer>();
        key.push(0);
        while(!key.isEmpty()){
            int now=key.pop();
            room.add(now);
            List<Integer> keys=rooms.get(now);
            for(int i=0;i<keys.size();i++){
                if(!room.contains(keys.get(i))){
                    key.push(keys.get(i));
                }
            }
        }
        if(room.size()==rooms.size()){
            return true;
        }else{
            return false;
        }
    }
}