LeetCode刷题记录——第892题（三维形体的表面积）

题目描述

在 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日 于燕南