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