每日一题-25

题目描述

三维形体的表面积
在 N * N 的网格上，我们放置一些 1 * 1 * 1 的立方体。

每个值 v = grid[i][j] 表示 v 个正方体叠放在对应单元格 (i, j) 上。

请你返回最终形体的表面积。

示例 1：

输入：[[2]]
输出：10
示例 2：

输入：[[1,2],[3,4]]
输出：34
示例 3：

输入：[[1,0],[0,2]]
输出：16
示例 4：

输入：[[1,1,1],[1,0,1],[1,1,1]]
输出：32
示例 5：

输入：[[2,2,2],[2,1,2],[2,2,2]]
输出：46

提示：

1 <= N <= 50
0 <= grid[i][j] <= 50

解题思路

• 不考虑侧边遮挡的情况时，每一个位置的表面积为v*4(侧边)+2(上下底面) ;v为此处叠放的立方体个数
• 遍历每一个网格，判断此位置与上左两个方向的立方体个数大小。减去上侧和左侧的遮挡的面积

代码

class Solution:
    def surfaceArea(self, grid: List[List[int]]) -> int:
        res = 0
        for i in range(len(grid)):
            for j in range(len(grid[0])):
                v = grid[i][j]
                if v>0:
                    res += v*4 + 2
                    res -= min(v, grid[i-1][j])*2 if i>0 else 0
                    res -= min(v, grid[i][j-1])*2 if j>0 else 0
        return res