【打卡】LeetCode 892. 三维形体的表面积

在 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

func GetMin(val1 int, val2 int) int {
    if val1 < val2 {
        return val1
    }
    return val2
}
func surfaceArea(grid [][]int) int {
    rows := len(grid)
    cols := len(grid[0])
    cnt := 0
 
    for i := 0; i < rows; i++ {
        for j := 0; j < cols; j++ {
            if grid[i][j] != 0 {

            begin := 6 + 4 * (grid[i][j] - 1)
            // 下方向
            if i + 1 < rows && grid[i + 1][j] != 0 {
                begin -= GetMin(grid[i][j], grid[i + 1][j]) * 2
            }
            // 右方向
            if j + 1 < cols && grid[i][j + 1] != 0 {
                begin -= GetMin(grid[i][j], grid[i][j + 1]) * 2
            }
            
            cnt += begin
            }
        }
    }

    return cnt
}