892. Surface Area of 3D Shapes

问题

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

类似题目

883. Projection Area of 3D Shapes