Leetcode 892.三维形体的表面积
1 题目描述(Leetcode题目链接)
在 N ∗ N N * N N∗N 的网格上,我们放置一些 1 ∗ 1 ∗ 1 1 * 1 * 1 1∗1∗1 的立方体。每个值 v = g r i d [ i ] [ j ] v = grid[i][j] v=grid[i][j] 表示 v 个正方体叠放在对应单元格 ( i , j ) (i, j) (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