Leetcode 892. 三维形体的表面积(Python3)

在 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: List[List[int]]) -> int:
        #n是立方体个数,m是接触面
        n,m = 0,0
        for i in range(len(grid)):
            for j in range(len(grid[0])):
                if grid[i][j]:
                    n += grid[i][j]
                if grid[i][j] > 1:
                    m += (grid[i][j] - 1) 
                #下
                if i < (len(grid) - 1) and grid[i+1][j]:
                    m += min(grid[i][j],grid[i+1][j])
                #右
                if j < (len(grid[0]) - 1) and grid[i][j+1]:
                    m += min(grid[i][j],grid[i][j+1])
        return (6 * n - 2 * m )

注：2019年秋招360