题目描述
在 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
思路分析
- 遍历grid的各个网格,首先将所有的六个面相加,如果在一个网格内堆叠的立方体大于1,那么就减去重叠的表面积;
- 相邻的网格中,存在高低不同的立方体,低的立方体决定了重叠的面积有多大,注意更新的规则
代码示例
class Solution(object):
def surfaceArea(self, grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
m = len(grid)
n = len(grid[0])
total = 0
for i in range(m):
for j in range(n):
total += grid[i][j] * 6
if grid[i][j] > 1:
total -= (grid[i][j] - 1) * 2
if j > 0:
total -= min(grid[i][j],grid[i][j-1]) * 2
if i > 0:
total -= min(grid[i][j],grid[i-1][j]) * 2
return total
2019年3月20日 于燕南