(87)841. 钥匙和房间(leetcode)

题目链接:
https://leetcode-cn.com/problems/keys-and-rooms/
难度:中等
841. 钥匙和房间
	有 N 个房间,开始时你位于 0 号房间。每个房间有不同的号码:0,1,2,...,N-1,
并且房间里可能有一些钥匙能使你进入下一个房间。
	在形式上,对于每个房间 i 都有一个钥匙列表 rooms[i],每个钥匙 rooms[i][j]
由 [0,1,...,N-1] 中的一个整数表示,其中 N = rooms.length。
 钥匙 rooms[i][j] = v 可以打开编号为 v 的房间。
	最初,除 0 号房间外的其余所有房间都被锁住。
	你可以自由地在房间之间来回走动。
	如果能进入每个房间返回 true,否则返回 false。
	
	1 <= rooms.length <= 1000
	0 <= rooms[i].length <= 1000
	所有房间中的钥匙数量总计不超过 3000。

不难 老方法 没什么好写的。。。
递归

class Solution {
    
    
public:
    int num;
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
    
    
        int n=rooms.size();
        vector<bool> flag(n,false);

        dfs(rooms,flag,0);

        return num==n;
    }

    void dfs(vector<vector<int>>& rooms,vector<bool>& flag,int i){
    
    
        flag[i]=true;
        num++;
        for(auto &a:rooms[i]){
    
    
            if(!flag[a]){
    
    
                dfs(rooms,flag,a);
            }
        }
    }
};

queue

class Solution {
    
    
public:
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
    
    
        int n=rooms.size();
        int num=0;
        vector<bool> flag(n,false);
        flag[0]=true;
        queue<int> que;
        que.push(0);
        while(!que.empty()){
    
    
            int i=que.front();
            que.pop();
            num++;
            for(auto &a:rooms[i]){
    
    
                if(!flag[a]){
    
    
                    flag[a]=true;
                    que.push(a);
                }
            }
        }

        return num==n;
    }

};

猜你喜欢

转载自blog.csdn.net/li_qw_er/article/details/108314008
今日推荐