问题

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

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

SRE实战 互联网时代守护先锋,助力企业售后服务体系运筹帷幄!一键直达领取阿里云限量特价优惠。

 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

扫码关注我们
微信号:SRE实战
拒绝背锅 运筹帷幄