【LeetCode】892. Surface Area of 3D Shapes

Problem

求3D图形的表面积。
二维数组v = grid[i][j],v表示图形的高,(i,j)表示所在的格子位置

Example

输入：[[2]]
输出：10
解释: 3D图形有4个侧面，面积都一样，两个上下面。S = 4 * h(高)+ 2 * 1
输入：[[1,2],[3,4]]
输出： 34
解释：每个的表面积为 4 * h + 2,总表面积为，4 * (1 + 2 + 3 + 4) + 4 * 2 = 48，再减去相邻位置重合的部分
[
[1,2],
[3,4]
]
第一行，1和2重合的面积为1 * 2 = 2,
第二行，3和4重合的面积为3 * 2 = 6，
第一列，1和3重合的面积为1 * 2 = 2，
第二列，2和4重合的面积为2 * 2 = 4
所以S = 48 - (2 + 6 + 2 + 4) = 34

Solution

class Solution {
    public int surfaceArea(int[][] grid) {
        int N= grid.length;
        int sum = 0;
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                sum += 4 * grid[i][j];
                if(grid[i][j] != 0) sum += 2;//这里特别注意一下，如果为0，就不算表面积了
                if(j < N - 1) {//如果为0，等式右边也为0，所以可以不用再加条件判断
                    sum -= 2 *Math.min(grid[i][j], grid[i][j + 1]);
                }
                if(i < N - 1) {
                    sum -= 2 * Math.min(grid[i][j], grid[i + 1][j]);
                }
            }
        }
        return sum;
    }
}

观察一下规律，理想状况下，总表面积为(4 * all_high(高度) + 2 * total(元素总个数，用来计算上下底面积))
每有一个相邻，就会减去较小那个高度的两倍，每次记录较小高度，最后在总和中- 2 * smaller_neighbor
每有一个0高度元素，就再减去上下两个面的面积，- 2 * zero
优化一下，把总高度，总个数，相邻数,高为0的，先计算出来，最后再处理。
S = (4 * all_high + 2 * total) - 2 * neighbor - 2 * zero

class Solution {
    public int surfaceArea(int[][] grid) {
        int N = grid.length;
        int total = N * N, zero = 0, smaller_neighbor = 0, all_high = 0;
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                int v = grid[i][j];
                if(v == 0) {
                    zero++;
                    continue;
                }
                all_high += v;
                if(j > 0) {
                    smaller_neighbor += Math.min(grid[i][j], grid[i][j - 1]);
                }
                if(i > 0) {
                    smaller_neighbor += Math.min(grid[i][j], grid[i - 1][j]);
                }
            }
        }
        return 4 * all_high + 2 * total - 2 * smaller_neighbor - 2 * zero;
    }
}