问题
NxN个格子中,用1x1x1的立方体堆叠,grid[i][j]表示坐标格上堆叠的立方体个数,求这个3D多边形的表面积。
Input: [[1,2],[3,4]]
Output: 34
思路
只要把每个柱体的表面积加起来(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