LeetCode 2257. Count Unguarded Cells in the Grid

🔗 leetcode.com/problems/co…

题目

• 给出一个 m x n 的二维表格，格子上有士兵 guard，有墙 wall
• 士兵可以盯上他上下左右所有的格子，碰到墙的格子就停止
• 返回没有被士兵盯到的格子的数量

思路

• 二维数组模拟处理即可，对于每一个士兵，不断占领四个方向的格子，直到碰到士兵，或者墙，就停止

• 最后统计没有被占领的格子的数量

• 看到过一个比较巧妙的四个方向遍历的预处理，即设置一个长度为 4 的数组，表达四个方向，四个方向可以表达为：

          dir.push_back({-1, 0});
          dir.push_back({0, -1});
          dir.push_back({1, 0});
          dir.push_back({0, 1});
  

代码

class Solution {
public:
    int countUnguarded(int m, int n, vector<vector<int>>& guards, vector<vector<int>>& walls) {
        // init
        bool mark[m][n], mark_wall[m][n];
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                mark[i][j] = mark_wall[i][j] = false;

            }
        }

        for (auto& wall : walls) {
            int r = wall[0], c = wall[1];
            mark_wall[r][c] = true;
            mark[r][c] = true;
        }

        for (auto& guard : guards) {
            int r = guard[0], c = guard[1];
            mark_wall[r][c] = true;
            mark[r][c] = true;
        }

        //iterator
        vector<vector<int>> dir;
        dir.push_back({-1, 0});
        dir.push_back({0, -1});
        dir.push_back({1, 0});
        dir.push_back({0, 1});

        for (auto& guard : guards) {
            int r = guard[0], c = guard[1];
            //printf("guard r %d c %d\n", r, c);
            for (int i = 0; i < dir.size(); i++) {
                int cur_r = r + dir[i][0], cur_c = c + dir[i][1];
                //printf("check r %d c %d\n", r, c);
                while (cur_r >= 0 && cur_r < m &&
                    cur_c >= 0 && cur_c < n && mark_wall[cur_r][cur_c] == false) {
                        mark[cur_r][cur_c] = true;
                        cur_r += dir[i][0];
                        cur_c += dir[i][1];
                        //printf("check r %d c %d\n", r, c);
                    }
            }
        }

        // count
        int ans = 0;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                if (mark[i][j] == false) ans++;
            }
        }

        return ans;

        
    }
};