leetcode892 三维形体表面积（python3）

leetcode892 三维形体表面积

在 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


直接上代码：

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

如果逐个计算每个单元格的面积的话很麻烦，那么我们就先计算出每个单元格所有面积，然后再减去相应的重叠部分，这样就简单很多了，注意减法不要重复计算。