Leetcode 892. 三维形体的表面积 python 版本

https://leetcode-cn.com/contest/weekly-contest-99/problems/surface-area-of-3d-shapes/

分为四步：

第一步：总面试是个数*6

第二步：同一位置，从2层以上开始，每升高一层，减少2个面

第三部：左右位置，每相邻一个，减少两个面

第四步：前后位置，每相邻一个，减少2个面

 

在 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

 

class Solution:
    def surfaceArea(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        rows = len(grid)
        columns = len(grid[0])
        sum_all = 0
        for i in range(rows):
            for j in range(columns):
                sum_all += grid[i][j]*6
                if grid[i][j] > 1:
                    sum_all -= (grid[i][j]-1)*2

                if j > 0:
                    sum_all -= min(grid[i][j], grid[i][j-1])*2
                if i > 0:
                    sum_all -= min(grid[i][j], grid[i-1][j])*2

        return sum_all