892. Surface Area of 3D Shapes

You are given an n x n grid where you have placed some 1 x 1 x 1 cubes. Each value v = grid[i][j] represents a tower of v cubes placed on top of cell (i, j).

After placing these cubes, you have decided to glue any directly adjacent cubes to each other, forming several irregular 3D shapes.

Return the total surface area of the resulting shapes.

Note: The bottom face of each shape counts toward its surface area.

 

Example 1:

Input: grid = [[2]]
Output: 10

Example 2:

Input: grid = [[1,2],[3,4]]
Output: 34

Example 3:

Input: grid = [[1,0],[0,2]]
Output: 16

Example 4:

Input: grid = [[1,1,1],[1,0,1],[1,1,1]]
Output: 32

Example 5:

Input: grid = [[2,2,2],[2,1,2],[2,2,2]]
Output: 46

 

Constraints:

• n == grid.length
• n == grid[i].length
• 1 <= n <= 50
• 0 <= grid[i][j] <= 50

Rust Solution

struct Solution;

impl Solution {
    fn surface_area(grid: Vec<Vec<i32>>) -> i32 {
        let n = grid.len();
        let mut res = 0;
        for i in 0..n {
            for j in 0..n {
                if grid[i][j] > 0 {
                    res += 2 + 4 * grid[i][j];
                }
            }
        }
        for i in 0..n {
            for j in 0..n {
                if i > 0 {
                    res -= 2 * i32::min(grid[i][j], grid[i - 1][j]);
                }
                if j > 0 {
                    res -= 2 * i32::min(grid[i][j], grid[i][j - 1]);
                }
            }
        }
        res
    }
}

#[test]
fn test() {
    let grid: Vec<Vec<i32>> = vec_vec_i32![[2]];
    assert_eq!(Solution::surface_area(grid), 10);
    let grid: Vec<Vec<i32>> = vec_vec_i32![[1, 2], [3, 4]];
    assert_eq!(Solution::surface_area(grid), 34);
    let grid: Vec<Vec<i32>> = vec_vec_i32![[1, 0], [0, 2]];
    assert_eq!(Solution::surface_area(grid), 16);
    let grid: Vec<Vec<i32>> = vec_vec_i32![[1, 1, 1], [1, 0, 1], [1, 1, 1]];
    assert_eq!(Solution::surface_area(grid), 32);
    let grid: Vec<Vec<i32>> = vec_vec_i32![[2, 2, 2], [2, 1, 2], [2, 2, 2]];
    assert_eq!(Solution::surface_area(grid), 46);
}