1. FlyPython首页
  2. 数据结构与算法
  3. leetcode题解

「LeetCode每日一题」——892. 三维形体的表面积

892. 三维形体的表面积

链接:https://leetcode-cn.com/problems/surface-area-of-3d-shapes/
难度:简单

题目

在 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

思路

这题目,没有一点想象力,还真读不懂。

来看个例子:

输入:[[2]]
输出:10

这个表示2个正方体叠着,10表示面积为10,因为总共12个面,有2个面接触被隐藏了。

再来看个例子:

输入:[[1,2],[3,4]]
输出:34

一列有1个正方体,2个叠加正方体,另外一列有3个叠加正方体,4个叠加的正方体。

最后看一下复杂的:

输入:[[2,2,2],[2,1,2],[2,2,2]]
输出:46

中间有空出来一个正方体,也是需要计算表面积的。

要算表面积无非是两个方法:
第一种,加法,把所有的面加起来
第二种,减法,把所有的立方体乘以6,再减去被遮蔽的面

在这里我们使用了第二种方法,代码见解决方案。

方案代码

解决方案:

class Solution:
    def surfaceArea(self, grid: List[List[int]]) -> int:
        # 计算相邻共同贴在一起的表面积
        def neighbour(r, c):
            sub_ngb = 0
            for nr, nc in ((r + 1, c), (r - 1, c), (r, c + 1), (r, c - 1)):
                if 0 <= nr < rows and 0 <= nc < cols:
                    sub_ngb += min(grid[r][c], grid[nr][nc])
            return sub_ngb
        ans = 0
        rows, cols = len(grid), len(grid[0])
        for r in range(rows):
            for c in range(cols):
                if grid[r][c] != 0:
                    # 不考虑贴在一起的表面积 = 4 * 高度(grid[r][c]) + 2
                    # 表面积 = 不考虑贴在一起的表面积 - 贴在一起的表面积
                    ans += 4 * grid[r][c] + 2 - neighbour(r, c)
        return ans

原创文章,作者:flypython,如若转载,请注明出处:http://flypython.com/algorithm/leetcode/254.html