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

posted @ 2018-10-22 12:24  PilgrimHui  阅读(476)  评论(0编辑  收藏  举报