【LeetCode】1001. 网格照明

题目

1001. 网格照明

在大小为 n x n 的网格 grid 上，每个单元格都有一盏灯，最初灯都处于 关闭 状态。

给你一个由灯的位置组成的二维数组 lamps ，其中 lamps[i] = [rowi, coli] 表示 打开 位于 grid[rowi][coli] 的灯。即便同一盏灯可能在 lamps 中多次列出，不会影响这盏灯处于 打开 状态。

当一盏灯处于打开状态，它将会照亮 自身所在单元格 以及同一 行 、同一 列 和两条 对角线 上的 所有其他单元格 。

另给你一个二维数组 queries ，其中 queries[j] = [rowj, colj] 。对于第 j 个查询，如果单元格 [rowj, colj] 是被照亮的，则查询结果为 1 ，否则为 0 。在第 j 次查询之后 [按照查询的顺序] ，关闭 位于单元格 grid[rowj][colj] 上及相邻 8 个方向上（与单元格 grid[rowi][coli] 共享角或边）的任何灯。

返回一个整数数组 ans 作为答案， ans[j] 应等于第 j 次查询 queries[j] 的结果，1 表示照亮，0 表示未照亮。

示例 1：

输入：n = 5, lamps = [[0,0],[4,4]], queries = [[1,1],[1,0]]
输出：[1,0]
解释：最初所有灯都是关闭的。在执行查询之前，打开位于 [0, 0] 和 [4, 4] 的灯。第 0 次查询检查 grid[1][1] 是否被照亮（蓝色方框）。该单元格被照亮，所以 ans[0] = 1 。然后，关闭红色方框中的所有灯。

第 1 次查询检查 grid[1][0] 是否被照亮（蓝色方框）。该单元格没有被照亮，所以 ans[1] = 0 。然后，关闭红色矩形中的所有灯。

示例 2：

输入：n = 5, lamps = [[0,0],[4,4]], queries = [[1,1],[1,1]]
输出：[1,1]

示例 3：

输入：n = 5, lamps = [[0,0],[0,4]], queries = [[0,4],[0,1],[1,4]]
输出：[1,1,0]

提示：

• 1 <= n <= 109
• 0 <= lamps.length <= 20000
• 0 <= queries.length <= 20000
• lamps[i].length == 2
• 0 <= rowi, coli < n
• queries[j].length == 2
• 0 <= rowj, colj < n

题解1（TLE）

思路

• 将所有的灯存在哈希表当中
• 对于query中的每一个元素，判断其八个方向是否有灯
• 判断过后再判断grid[rowj][colj]是否有灯，有则删除

代码

from typing import List


class Solution:
    # TLE
    def gridIllumination(self, n: int, lamps: List[List[int]], queries: List[List[int]]) -> List[int]:
        res = []
        lampSet = set()
        for x, y in lamps:
            lampSet.add((x, y))
        directions = [(1, 1), (1, 0), (1, -1), (0, 1),
                      (0, 0), (0, -1), (-1, 1), (-1, 0), (-1, -1)]
        for x, y in queries:
            flag = 0
            for x0, y0 in lampSet:
                if x == x0 or y == y0 or abs(x-x0) == abs(y-y0):
                    flag = 1
            if flag == 1:
                res.append(1)
                flag = 0
            else:
                res.append(0)

            for dx, dy in directions:
                if (x+dx, y + dy) in lampSet:
                    lampSet.remove((x+dx, y + dy))
        return res

复杂度

• 时间复杂度：$O(l\times q)$
• 空间复杂度：$O(l+q)$

题解2

思路

• 将灯的所在行、列、对角线、反对角线标注
• 判断该点是否发光只需要判断该点是否在行、列、对角线、反对角线上即可

代码

from collections import Counter
from typing import List


class Solution:
    def gridIllumination(self, n: int, lamps: List[List[int]], queries: List[List[int]]) -> List[int]:
        res = []
        lampSet, row, col, diagonal, antidiagonal = set(), Counter(), Counter(), Counter(), Counter()
        directions = [(1, 1), (1, 0), (1, -1), (0, 1),
                      (0, 0), (0, -1), (-1, 1), (-1, 0), (-1, -1)]
        for x, y in lamps:
            if (x,y) not in lampSet:
                lampSet.add((x, y))
                row[x] += 1
                col[y] += 1
                diagonal[x+y] += 1
                antidiagonal[x-y] += 1

        for x, y in queries:
            if row[x] or col[y] or diagonal[x + y] or antidiagonal[x-y]:
                res.append(1)
            else:
                res.append(0)
            for dx, dy in directions:
                x0, y0 = x+dx, y+dy
                if (x0, y0) in lampSet:
                    lampSet.remove((x0, y0))
                    row[x0] -= 1
                    col[y0] -= 1
                    diagonal[x0 + y0] -= 1
                    antidiagonal[x0-y0] -= 1
        return res

复杂度

• 时间复杂度：$O(l)$
• 空间复杂度：$O(l+q)$