携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第25天,点击查看活动详情
一、题目描述:
892. 三维形体的表面积 - 力扣(LeetCode) (leetcode-cn.com)
给你一个 n * n 的网格 grid ,上面放置着一些 1 x 1 x 1 的正方体。每个值 v = grid[i][j] 表示 v 个正方体叠放在对应单元格 (i, j) 上。
放置好正方体后,任何直接相邻的正方体都会互相粘在一起,形成一些不规则的三维形体。
请你返回最终这些形体的总表面积。
注意:每个形体的底面也需要计入表面积中。
示例 1:
输入:grid = [[1,2],[3,4]]
输出:34
示例 2:
输入:grid = [[1,1,1],[1,0,1],[1,1,1]]
输出:32
示例 3:
输入:grid = [[2,2,2],[2,1,2],[2,2,2]]
输出:46
提示:
- n == grid.length
- n == grid[i].length
- 1 <= n <= 50
- 0 <= grid[i][j] <= 50
二、思路分析:
对于每个格子, 如果里面有方块, 上表面积和下表面积都+1
然后判断他周围的4个位置是否有格子:
如果没有, 加上他格子的个数
如果有且比他高, 那么不加任何值
如果有且比他低, 那么加上他们数量的差值
三、AC 代码:
class Solution:
def surfaceArea(self, grid: List[List[int]]) -> int:
"""
前
左u/d 右
后
"""
if not grid:
return 0
m, n = len(grid), len(grid[0])
res = 0
for i in range(m):
for j in range(n):
# 上下两个面
if grid[i][j]: res += 2
# 前
if i == 0:
res += grid[i][j]
else:
tmp = grid[i][j] - grid[i - 1][j]
if tmp > 0: res += tmp
# 后
if i == m - 1:
res += grid[i][j]
else:
tmp = grid[i][j] - grid[i + 1][j]
if tmp > 0: res += tmp
# 左
if j == 0:
res += grid[i][j]
else:
tmp = grid[i][j] - grid[i][j - 1]
if tmp > 0: res += tmp
# 右
if j == n - 1:
res += grid[i][j]
else:
tmp = grid[i][j] - grid[i][j + 1]
if tmp > 0: res += tmp
return res