糖逗 发表于 2020-6-2 10:30:53

C++刷leetcode(841. 钥匙和房间)【深度优先搜索】【图】

题目描述:
有 N 个房间,开始时你位于 0 号房间。每个房间有不同的号码:0,1,2,...,N-1,并且房间里可能有一些钥匙能使你进入下一个房间。

在形式上,对于每个房间 i 都有一个钥匙列表 rooms,每个钥匙 rooms 由 中的一个整数表示,其中 N = rooms.length。 钥匙 rooms = v 可以打开编号为 v 的房间。

最初,除 0 号房间外的其余所有房间都被锁住。

你可以自由地在房间之间来回走动。

如果能进入每个房间返回 true,否则返回 false。

示例 1:

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

输入:[,,,]
输出:false
解释:我们不能进入 2 号房间。
提示:

1 <= rooms.length <= 1000
0 <= rooms.length <= 1000
所有房间中的钥匙数量总计不超过 3000。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/keys-and-rooms
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

class Solution {
public:
    void dfs(vector<vector<int> >& rooms, vector<int>&temp, vector<int>&visit, bool&res, int cur_room){         
      if(temp.size() == rooms.size()){
            res = true;
            return;
      }
      int store = 0;
      for(int i = 0; i < rooms.size(); i++){
            if(visit] == 1) store++;
      }
      if(store == rooms.size()) return;
      for(int i = 0 ; i < rooms.size(); i++){
            if(visit] == 0){
                temp.push_back(rooms);
                visit] = 1;
                dfs(rooms, temp, visit, res, rooms);
            }
      }
    }
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
      vector<int>visit(rooms.size(), 0);
      bool res = false;
      vector<int> temp;
      visit = 1;
      temp.push_back(0);
      dfs(rooms, temp, visit, res, 0);
      return res;
    }
};

糖逗 发表于 2020-6-2 10:31:35

{:10_327:}
页: [1]
查看完整版本: C++刷leetcode(841. 钥匙和房间)【深度优先搜索】【图】