Python Leetcode(892.三维形体的表面积)

Python 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



 Solution:(类似于之前计算岛屿周长，先计算出每个点小方块堆积起来的面积，之后为了不重复计算，对每个点的左方和上方进行检查有无相邻方块，若有，则减去数量少的那个方块的个数乘2.)   

class Solution(object):
    def surfaceArea(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        result = 0
        x, y = len(grid), len(grid[0])
        
        for i in range(x):
            for j in range(y):
                if grid[i][j] != 0:
                    result += grid[i][j] * 6 - (grid[i][j] - 1) * 2
                if i > 0 and grid[i-1][j] != 0:
                    result -= min(grid[i-1][j], grid[i][j]) * 2
                if j > 0 and grid[i][j-1] != 0:
                    result -= min(grid[i][j-1], grid[i][j]) * 2
        return result

solution = Solution()
print(solution.surfaceArea([[1,1,1],[1,0,1],[1,1,1]]))

32