Leetcode 883.三维形体投影面积
1 题目描述(Leetcode题目链接)
在 N * N 的网格中,我们放置了一些与 x,y,z 三轴对齐的 1 * 1 * 1 立方体。
每个值 v = grid[i][j] 表示 v 个正方体叠放在单元格 (i, j) 上。
现在,我们查看这些立方体在 xy、yz 和 zx 平面上的投影。
投影就像影子,将三维形体映射到一个二维平面上。
在这里,从顶部、前面和侧面看立方体时,我们会看到“影子”。
返回所有三个投影的总面积。
输入:[[1,2],[3,4]]
输出:17
解释:
这里有该形体在三个轴对齐平面上的三个投影(“阴影部分”)。
提示:
- 1 <= grid.length = grid[0].length <= 50
- 0 <= grid[i][j] <= 50
2 题解
面积=非零元素的个数+每行的最大值+每列的最大值
class Solution:
def projectionArea(self, grid: List[List[int]]) -> int:
m, n = len(grid), len(grid)
row = [0]*m
col = [0]*n
res = 0
for i in range(m):
for j in range(n):
if grid[i][j] != 0:
res += 1
row[i] = max(row[i], grid[i][j])
col[j] = max(col[j], grid[i][j])
return res + sum(row) + sum(col)