892. Surface Area of 3D Shapes

时间:2023-03-10 07:08:23
892. Surface Area of 3D Shapes

问题

NxN个格子中,用1x1x1的立方体堆叠,grid[i][j]表示坐标格上堆叠的立方体个数,求这个3D多边形的表面积。

Input: [[1,2],[3,4]]

Output: 34

892. Surface Area of 3D Shapes

思路

只要把每个柱体的表面积加起来(grid[i][j] * 4 ,4表示四个侧面,2表示上下两个面),然后减去重叠的部分即可。

重叠的部分为x方向(或y方向)上相邻柱体中较小的grid值。

时间复杂度O(n^2),空间复杂度O(1)

代码

class Solution(object):
def surfaceArea(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
s = 0
n = len(grid)
for i in range(n):
for j in range(n):
if grid[i][j]:
s += grid[i][j] * 4 + 2
if i:
s -= min(grid[i][j], grid[i-1][j]) * 2
if j:
s -= min(grid[i][j], grid[i][j-1]) * 2
return s

类似题目

883. Projection Area of 3D Shapes