欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

[LeetCode] 892、三维形体的表面积

程序员文章站 2024-03-15 11:35:41
...

题目描述

N * N 的网格上,我们放置一些 1 * 1 * 1 的立方体。每个值 v = grid[i][j] 表示 v 个正方体叠放在对应单元格 (i, j) 上。请你返回最终形体的表面积。

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

[LeetCode] 892、三维形体的表面积

解题思路

关键是要先理解题意。首先,一个柱体一个柱体的看,每个柱体是由:2个底面(上表面/下表面)+ 所有的正方体都贡献了4个侧表面积。然后,把柱体贴合在一起之后,我们需要把贴合的表面积给减掉,两个柱体贴合的表面积就是:两个柱体高的最小值*2

参考代码

class Solution {
public:
    int surfaceArea(vector<vector<int>>& grid) {
        int n = grid.size(), area = 0;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                // 先把grid[i][j]赋值给level,省掉了bound check,可以略微略微略微优化一下耗时。。。
                int level = grid[i][j];
                if (level > 0) {
                    // 一个柱体中:2个底面 + 所有的正方体都贡献了4个侧表面积 
                    area += (level << 2) + 2;
                    // 减掉 i 与 i-1 相贴的两份表面积
                    area -= i > 0? min(level, grid[i - 1][j]) << 1: 0; 
                    // 减掉 j 与 j-1 相贴的两份表面积
                    area -= j > 0? min(level, grid[i][j - 1]) << 1: 0;
                }  
            }
        }
        return area;
    }
};