README.md

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

English Version

题目描述

给你两个整数 m 和 n 表示一个下标从 0 开始的 m x n 网格图。同时给你两个二维整数数组 guards 和 walls ，其中 guards[i] = [rowi, coli] 且 walls[j] = [rowj, colj] ，分别表示第 i 个警卫和第 j 座墙所在的位置。

一个警卫能看到 4 个坐标轴方向（即东、南、西、北）的 所有 格子，除非他们被一座墙或者另外一个警卫 挡住 了视线。如果一个格子能被 至少 一个警卫看到，那么我们说这个格子被 保卫 了。

请你返回空格子中，有多少个格子是 没被保卫 的。

 

示例 1：

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

示例 2：

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

 

提示：

• 1 <= m, n <= 105
• 2 <= m * n <= 105
• 1 <= guards.length, walls.length <= 5 * 104
• 2 <= guards.length + walls.length <= m * n
• guards[i].length == walls[j].length == 2
• 0 <= rowi, rowj < m
• 0 <= coli, colj < n
• guards 和 walls 中所有位置 互不相同 。

解法

Python3

class Solution:
    def countUnguarded(
        self, m: int, n: int, guards: List[List[int]], walls: List[List[int]]
    ) -> int:
        g = [[None] * n for _ in range(m)]
        for r, c in guards:
            g[r][c] = 'g'
        for r, c in walls:
            g[r][c] = 'w'
        for i, j in guards:
            for a, b in [[0, -1], [0, 1], [1, 0], [-1, 0]]:
                x, y = i, j
                while (
                    0 <= x + a < m
                    and 0 <= y + b < n
                    and g[x + a][y + b] != 'w'
                    and g[x + a][y + b] != 'g'
                ):
                    x, y = x + a, y + b
                    g[x][y] = 'v'
        return sum(not v for row in g for v in row)

Java

class Solution {
    public int countUnguarded(int m, int n, int[][] guards, int[][] walls) {
        char[][] g = new char[m][n];
        for (int[] e : guards) {
            int r = e[0], c = e[1];
            g[r][c] = 'g';
        }
        for (int[] e : walls) {
            int r = e[0], c = e[1];
            g[r][c] = 'w';
        }
        int[][] dirs = new int[][] {{0, -1}, {0, 1}, {1, 0}, {-1, 0}};
        for (int[] p : guards) {
            for (int[] dir : dirs) {
                int a = dir[0], b = dir[1];
                int x = p[0], y = p[1];
                while (x + a >= 0 && x + a < m && y + b >= 0 && y + b < n && g[x + a][y + b] != 'w'
                    && g[x + a][y + b] != 'g') {
                    x += a;
                    y += b;
                    g[x][y] = 'v';
                }
            }
        }
        int ans = 0;
        for (char[] row : g) {
            for (char v : row) {
                if (v == 0) {
                    ++ans;
                }
            }
        }
        return ans;
    }
}

C++

class Solution {
public:
    int countUnguarded(int m, int n, vector<vector<int>>& guards, vector<vector<int>>& walls) {
        vector<vector<char>> g(m, vector<char>(n));
        for (auto& e : guards) g[e[0]][e[1]] = 'g';
        for (auto& e : walls) g[e[0]][e[1]] = 'w';
        vector<vector<int>> dirs = {{0, 1}, {0, -1}, {1, 0}, {-1, 0}};
        for (auto& p : guards) {
            for (auto& dir : dirs) {
                int a = dir[0], b = dir[1];
                int x = p[0], y = p[1];
                while (x + a >= 0 && x + a < m && y + b >= 0 && y + b < n && g[x + a][y + b] != 'w' && g[x + a][y + b] != 'g') {
                    x += a;
                    y += b;
                    g[x][y] = 'v';
                }
            }
        }
        int ans = 0;
        for (auto& row : g)
            for (auto& v : row)
                ans += v == 0;
        return ans;
    }
};

Go

func countUnguarded(m int, n int, guards [][]int, walls [][]int) int {
	g := make([][]int, m)
	for i := range g {
		g[i] = make([]int, n)
	}
	for _, e := range guards {
		g[e[0]][e[1]] = 1
	}
	for _, e := range walls {
		g[e[0]][e[1]] = 2
	}
	dirs := [][]int{{0, -1}, {0, 1}, {1, 0}, {-1, 0}}
	for _, p := range guards {
		for _, dir := range dirs {
			a, b := dir[0], dir[1]
			x, y := p[0], p[1]
			for x+a >= 0 && x+a < m && y+b >= 0 && y+b < n && g[x+a][y+b] != 1 && g[x+a][y+b] != 2 {
				x, y = x+a, y+b
				g[x][y] = 3
			}
		}
	}
	ans := 0
	for _, row := range g {
		for _, v := range row {
			if v == 0 {
				ans++
			}
		}
	}
	return ans
}

TypeScript

...