Leetcode 892.三维形体的表面积（Surface Area of 3D Shapes）

Leetcode 892.三维形体的表面积

1 题目描述（Leetcode题目链接）

  在 $N * N$ 的网格上，我们放置一些 $1 * 1 * 1$ 的立方体。每个值 $v = grid[i][j]$ 表示 v 个正方体叠放在对应单元格 $(i, j)$ 上。请你返回最终形体的表面积。

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

输入：[[1,1,1],[1,0,1],[1,1,1]]
输出：32

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

2 题解

  题目的意思其实就是在平面网格上堆立方体，最后求一下这些立方体堆成物体的表面积。那么挨个加就好了，遍历到每个格子的时候先加2（上下两个面），再向四周判断，只要当前位置比周围位置高，那就加上当前位置与周围位置的差。画个图想象一下就很好理解了。

class Solution:
    def surfaceArea(self, grid: List[List[int]]) -> int:
        retv = 0
        n = len(grid)
        di = [(0, 1), (1, 0), (0, -1), (-1, 0)]
        for i in range(n):
            for j in range(n):
                if grid[i][j] > 0:
                    retv += 2
                    for d in di:
                        x, y = i + d[0], j + d[1]
                        if x >= 0 and x < n and y >= 0 and y < n:
                            retv += max(grid[i][j]-grid[x][y], 0)
                        else:
                            retv += grid[i][j]
        return retv