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：

輸入: [[1],[2],[3],[]]
輸出: true
解釋:
我們從 0 號房間開始，拿到鑰匙 1。
之後我們去 1 號房間，拿到鑰匙 2。
然後我們去 2 號房間，拿到鑰匙 3。
最後我們去了 3 號房間。
由於我們能夠進入每個房間，我們返回 true。
示例 2：

輸入：[[1,3],[3,0,1],[2],[0]]
輸出：false
解釋：我們不能進入 2 號房間。
提示：

1 <= rooms.length <= 1000
0 <= rooms[i].length <= 1000
所有房間中的鑰匙數量總計不超過 3000。

AC程式碼：

class Solution {
public:
    int f[1010];
    int num = 0;
    
    void getrooms(vector<vector<int>>& rooms,int id)
    {
        int m = rooms[id].size();
        for(int i=0;i<m;i++)
        {
            if(f[rooms[id][i]] == 0)
            {
                f[rooms[id][i]] = 1;
                num++;
                getrooms(rooms,rooms[id][i]);
            }
        }
    }    
    
    bool canVisitAllRooms(vector<vector<int>>& rooms) {
        int n = rooms.size();
        if(n == 1) return true;
        memset(f,0,sizeof(f));
        f[0] = 1;
        num = 1;
        getrooms(rooms,0);
        if(num == n) return true;
        return false;
    }
};