Leetcode笔记 每日一题 883. 三维形体投影面积(22.04.26)

Leetcode笔记 每日一题 883. 三维形体投影面积(22.04.26)

Leetcode 每日一题 883. 三维形体投影面积(22.04.26)

题目

在 n x n 的网格 grid 中，我们放置了一些与 x，y，z 三轴对齐的 1 x 1 x 1 立方体。

每个值 v = grid[i][j] 表示 v 个正方体叠放在单元格 (i, j) 上。

现在，我们查看这些立方体在 xy 、yz 和 zx 平面上的投影。

投影 就像影子，将 三维 形体映射到一个 二维 平面上。从顶部、前面和侧面看立方体时，我们会看到“影子”。

返回 所有三个投影的总面积 。

示例1

输入：[[1,2],[3,4]]
输出：17
解释：这里有该形体在三个轴对齐平面上的三个投影(“阴影部分”)。

示例 2:

输入：grid = [[2]]
输出：5

示例 3：

输入：[[1,0],[0,2]]
输出：8

提示：

n == grid.length == grid[i].length
1 <= n <= 50
0 <= grid[i][j] <= 50

解题思路

从题目可以看出，这里主要求的是三个视图的面积之和：分别是主视图(x-z)、侧视图(y-z)和俯视图(x-y)。所以只要分别算出这三个俯视图的面积，最后相加便可。而x、y、z分别的是len(grid)、len(grid[x])和grid[x][y]
三个视图面积计算：

• 俯视图(x-y)： 通过图片可以知道，只要grid[x][y]>0即平面x-y上存在正方体，面积s+=1
• 主视图(x-z)：根据透视规律，只要考虑x轴上z的最大值即可，即s+= max(grid[x])
• 侧视图(y-z)：与主视图原理相同，只是x轴变成y轴，所以可以用zip()对grid进行转置，即zip(*grid)，最后计算面积s += max(grid[y])

Python代码

class Solution:
    def projectionArea(self, grid: List[List[int]]) -> int:
        s = 0 
        n = len(grid)
        # 俯视图计算，先判断平面x-y上是否存在正方体
        for x in range(n):
            for y in range(len(grid[x])):
                if grid[x][y] > 0:
                    s += 1
        # 主视图计算
        for x in range(n):
            s += max(grid[x])
        # 对grid进行转置
        grid_y = list(zip(*grid))
        # 侧视图计算
        for y in range(len(grid_y)):
            s += max(grid_y[y])
        return s

简便算法

class Solution:
    def projectionArea(self, grid: List[List[int]]) -> int:
        s = sum(v > 0 for i in grid for v in i) # 用题目中v=grid[i][j]表示坐标轴x-y上的正方体
        s += sum(map(max, zip(*grid)))  # 注意这里为 O(n) 空间复杂度，改为下标枚举则可以 O(1)
        s += sum(map(max, grid))
        return s