题目
在 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
来源:力扣(LeetCode)
解题思路
很简单的观察法,从底部看只要数组里遇到一个不为0的数都可以记一个面积,从x轴看去就是列表里每个元素的最大值加起来,从y轴看就是矩阵每列的最大值相加。
class Solution:
def projectionArea(self, grid: List[List[int]]) -> int:
bottom=len(grid)*len(grid[0])
x=0
zero=0
for i in grid:
zero+=i.count(0) #统计0的个数
x+=max(i)
y=0
for j in range(len(grid[0])):
y+=max([i[j] for i in grid])
return bottom+x+y-zero