LeetCode 2257. 统计网格图中没有被保卫的格子数

输入：m = 4, n = 6, 
guards = [[0,0],[1,1],[2,3]], 
walls = [[0,1],[2,2],[1,4]]
输出：7
解释：上图中，被保卫和没有被保卫的格子分别用红色和绿色表示。
总共有 7 个没有被保卫的格子，所以我们返回 7 。

输入：m = 3, n = 3, 
guards = [[1,1]], 
walls = [[0,1],[1,0],[2,1],[1,2]]
输出：4
解释：上图中，没有被保卫的格子用绿色表示。
总共有 4 个没有被保卫的格子，所以我们返回 4 。
 
提示：
1 <= m, n <= 10^5
2 <= m * n <= 10^5
1 <= guards.length, walls.length <= 5 * 10^4
2 <= guards.length + walls.length <= m * n
guards[i].length == walls[j].length == 2
0 <= rowi, rowj < m
0 <= coli, colj < n
guards 和 walls 中所有位置 互不相同 。

class Solution {
public:
    int countUnguarded(int m, int n, vector<vector<int>>& guards, vector<vector<int>>& walls) {
        vector<vector<int>> g(m, vector<int>(n,0));
        vector<vector<int>> dir = {{-1, 0}, {0, -1}, {1, 0}, {0, 1}};
        int police = 1, wall = 2, mark = 3;
        for(auto& p : guards)
            g[p[0]][p[1]] = police;
        for(auto& w : walls)
            g[w[0]][w[1]] = wall;
        for(int i = 0; i < m; ++i)
        {
            for(int j = 0; j < n; ++j)
            {
                if(g[i][j]==police)
                {    
                    for(int k = 0; k < 4; ++k)
                    {
                        int nx = i, ny = j;
                        while(1)
                        {
                            nx += dir[k][0], ny += dir[k][1];
                            if(nx>=0 && nx<m && ny>=0 && ny<n)
                            {
                                if(g[nx][ny]==police || g[nx][ny]==wall)
                                    break;
                                g[nx][ny] = mark;
                            }
                            else
                                break;
                        }
                    }
                }
            }
        }
        int ans = 0;
        for(int i = 0; i < m; ++i)
        {
            for(int j = 0; j < n; ++j)
            {
                if(g[i][j]==0)
                    ans++;
            }
        }
        return ans;
    }
};