LeetCode刷题之883. 三维形体投影面积
我不知道将去向何方,但我已在路上! |
---|
时光匆匆,虽未曾谋面,却相遇于斯,实在是莫大的缘分,感谢您的到访 ! |
- 题目:
在N*N
的网格中,我们放置了一些与 x,y,z
三轴对齐的1 * 1 * 1
立方体。每个值v = grid[i][j]
表示v
个正方体叠放在单元格(i, j)
上。现在,我们查看这些立方体在 xy
、yz
和 zx
平面上的投影。投影就像影子,将三维形体映射到一个二维平面上。在这里,从顶部、前面和侧面看立方体时,我们会看到“影子”。返回所有三个投影的总面积 - 示例:
示例 1 :
输入:[[2]]
输出:5
示例 2 :
输入:[[1,2],[3,4]]
输出:17
解释:这里有该形体在三个轴对齐平面上的三个投影(“阴影部分”)。
示例 3:
输入:[[1,0],[0,2]]
输出:8
示例 4:
输入:[[1,1,1],[1,0,1],[1,1,1]]
输出:14
示例 5:
输入:[[2,2,2],[2,1,2],[2,2,2]]
输出:21
- 1 <=
grid.length
= grid[0].length
<= 50 - 0 <=
grid[i][j]
<= 50
class Solution:
def projectionArea(self, grid: List[List[int]]) -> int:
a = []
count_xz,count_yz = 0,0
for i in range(len(grid)):
count_xz += max(grid[i])
a.extend(grid[i])
count_xy = len(a) - a.count(0)
for j in range(len(grid)):
count_yz += max(a[j::len(grid)])
return(count_xy + count_xz + count_yz)
- 算法说明:
分为俯视图、前视图、左视图三部分进行计算:
(1)俯视图(xy):检查每一个柱子里面是否堆放了方块,也就是检查是否有元素为 0;
(2)前视图(xz):求出每行的最大值;
(3)左视图(yz):求出每列的最大值;
程序运行过程: 按行进行遍历grid
元素,将每一行元素的最大值累加到前视图的面积值;并将每一行的元素用extend()
函数添加到列表a
中,在第一个for
循环结束后,用a
中所有元素个数减去0
元素的个数,求出俯视图的面积值;求出每列的最大值累加到左视图的面积上,再求每一列的最大值时,在列表a
中采用间隔法,逐个选取列元素,每间隔len(grid)
的长度选择一个元素,刚好是按列选择元素。最后返回三个视图面积的和。