leetcode--892. 三维形体的表面积

  1. 三维形体的表面积
    在 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]]
[[1,2],[3,4]]
[[1,0],[0,2]]
[[1,1,1],[1,0,1],[1,1,1]]
[[2,2,2],[2,1,2],[2,2,2]]
[[0,3,4,3],[4,5,0,5],[0,4,2,4],[4,0,0,2]]

class Solution {
public:
    int surfaceArea(vector<vector<int>>& grid) {
        int n = grid.size();
        if (0 == n) return 0;
        int ans = grid[0][0] == 0 ? 0 : grid[0][0] * 6 - 2 * (grid[0][0] - 1);
        for (int j=1; j<n; j++) {
            if (0 != grid[0][j]) {
                int temp = grid[0][j-1] < grid[0][j] ? grid[0][j-1] : grid[0][j];
                ans += grid[0][j] * 6 - 2 * (grid[0][j] - 1) - temp * 2;
            }
        }
        for (int i=1; i<n; i++) {
            for (int j=0; j<n; j++) {
                if (0 != grid[i][j]) {
                    int temp1 = grid[i-1][j] < grid[i][j] ? grid[i-1][j] : grid[i][j];
                    if (0 == j) {
                        ans += grid[i][j] * 6 - 2 * (grid[i][j] - 1) - temp1 * 2;
                    } else {
                        int temp2 = grid[i][j-1] < grid[i][j] ? grid[i][j-1] : grid[i][j];
                        ans += grid[i][j] * 6 - 2 * (grid[i][j] - 1) - temp1 * 2 - temp2 * 2;
                    }
                }
            }
        }
        return ans;
    }
};
/*32ms,8.8MB*/

时间复杂度:O(N^2)
空间复杂度:O(1)

优化版:

class Solution {
public:
    int surfaceArea(vector<vector<int>>& grid) {
        int n = grid.size();
        int ans = 0;
        for (int i=0; i<n; i++) {
            for (int j=0; j<n; j++) {
                int cur = grid[i][j];
                if (cur > 0) {
                    ans += (cur << 2) + 2;
                    ans -= i > 0 ? min(cur, grid[i-1][j]) << 1 : 0;
                    ans -= j > 0 ? min(cur, grid[i][j-1]) << 1 : 0;
                }
            }
        }
        return ans;
    }
};
/*16ms,9MB*/

参考链接:
https://leetcode-cn.com/problems/surface-area-of-3d-shapes/solution/shi-li-you-tu-you-zhen-xiang-jiang-jie-yi-kan-jiu-/

发布了59 篇原创文章 · 获赞 0 · 访问量 1179

猜你喜欢

转载自blog.csdn.net/u011861832/article/details/105089621
今日推荐