有 n 个房间,房间按从 0 到 n - 1 编号。最初,除 0 号房间外的其余所有房间都被锁住。
你的目标是进入所有的房间。然而,你不能在没有获得钥匙的时候进入锁住的房间。
当你进入一个房间,你可能会在里面找到一套不同的钥匙,
每把钥匙上都有对应的房间号,即表示钥匙可以打开的房间。
你可以拿上所有钥匙去解锁其他房间。
给你一个数组 rooms 其中 rooms[i] 是你进入 i 号房间可以获得的钥匙集合。
如果能进入 所有 房间返回 true,否则返回 false
解题思路:
题目的意思就是一个单向图问题,只要从0号房深度或者广度遍历到所有结点,说明可行
遍历不到所有结点,说明存在孤立结点||没有钥匙的房门
class Solution(object):
def canVisitAllRooms(self, rooms):
# 广度优先
roomsNum = len(rooms)
num = 0
visited = {0}
queue = collections.deque([0])
while queue:
nextKey = queue.popleft()
num += 1
for door in rooms[nextKey]:
if door not in visited:
visited.add(door)
queue.append(door)
return num == roomsNum