leetcode 883. 三维形体投影面积(python)

题目链接

题目描述:

在 N * N 的网格中,我们放置了一些与 x,y,z 三轴对齐的 1 * 1 * 1 立方体。

每个值 v = grid[i][j] 表示 v 个正方体叠放在单元格 (i, j) 上。

现在,我们查看这些立方体在 xy、yz 和 zx 平面上的投影。

投影就像影子,将三维形体映射到一个二维平面上。

在这里,从顶部、前面和侧面看立方体时,我们会看到“影子”。

返回所有三个投影的总面积。

示例 1:

输入:[[2]]
输出:5
示例 2:

输入:[[1,2],[3,4]]
输出:17
解释:
这里有该形体在三个轴对齐平面上的三个投影(“阴影部分”)。
在这里插入图片描述

示例 3:

输入:[[1,0],[0,2]]
输出:8
示例 4:

输入:[[1,1,1],[1,0,1],[1,1,1]]
输出:14
示例 5:

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

解题思路:
grid的长度表示立方体的行数,中间每一个元素表示叠放的立方体个数,x-y上的阴影面积就是grid中不为0的个数,其余两个方向的投影分别为每行最大值之和,每列最大值只和。

class Solution:
    def projectionArea(self, grid: List[List[int]]) -> int:
        area=0
        for i in grid:
            area+=max(i)
            for j in i:
                if j!=0:
                    area+=1
        for i in range(0,len(grid[0])):
            mm=0
            for j in range(len(grid)):
                if grid[j][i]>mm:
                    mm= grid[j][i]
            area+=mm
        return area

p.s.别人的做法
计算x-y投影时,不用遍历每一个元素,遍历每一行统计这一行0的个数,用元素总数减去即可;用zip(*grid)得到转置后的矩阵,这样在统计每一行的最大值,用一重循环即可。

        sum=0
        for row in grid:
            sum+=max(row)
            sum-=row.count(0)
        for col in zip(*grid):
            sum+=max(col)
        return sum+len(grid)*len(grid)

猜你喜欢

转载自blog.csdn.net/weixin_44740082/article/details/88877048