883. Projection Area of 3D Shapes

EasyGeometryArrayMathMatrix
Leetcode Link

Problem Description

In this problem, we are given an n x n grid where each cell in the grid contains a non-negative integer value representing the height of a tower of cubes placed in that cell. The task is to calculate the total surface area of all three orthogonal projections (top-down, front, and side) of the arrangement of the cubes.

To understand this better, imagine looking at the cubes from three different viewpoints: from directly above (the xy-plane), from the front (the yz-plane), and from the side (the zx-plane). The area of the projection is simply the shadow of these cubes on the respective planes. The top-down view counts all the cubes that are visible from above, the front view counts the tallest tower in each row, and the side view counts the tallest tower in each column.

Intuition

For the solution, we can tackle each projection separately and sum up their areas:

  • xy-plane (Top-Down View): To compute the area of the projection on the xy-plane, we count the number of cubes that are visible from the top. Each cube contributes to the area if its height is greater than zero. We simply iterate over the grid and count the number of cells with a value greater than zero.

  • yz-plane (Front View): For the yz-plane, we are looking at the grid from the front, so we need to consider the tallest tower in each row since only the tallest tower is visible from this viewpoint. We iterate through each row and take the maximum value of the row, representing the height of the tallest tower.

  • zx-plane (Side View): Similarly, for the zx-plane projection, which represents the view from the side, we need to find the tallest tower in each column. We can achieve this by zipping the columns together and finding the maximum height of each column.

The final area of all projections is the sum of the three calculated areas: the top-down, the front, and the side projection areas.

Learn more about Math patterns.

Solution Approach

The implementation of the solution uses simple yet effective techniques to calculate the projection areas on all three planes. The choice of data structures is straightforward - since the input is a grid (a list of lists in Python), we use this same structure to operate on the data. Here's how the solution is structured:

  • xy-plane (Top-Down View): The xy variable is calculated by using a nested list comprehension. The outer loop iterates over each row, and the inner loop iterates over each value in the row, checking if v > 0. This check effectively counts the number of cells with cubes visible from the top view, contributing to the xy area. The use of list comprehension makes this part compact and efficient.

    xy = sum(v > 0 for row in grid for v in row)  
  • yz-plane (Front View): The yz projection is calculated by iterating over each row in the grid and selecting the maximum value found in that row using the max function. This represents the highest tower when looking from the front, and the sum of these maximum values across all rows gives the yz area.

    yz = sum(max(row) for row in grid)
  • zx-plane (Side View): To find the zx projection, we need the tallest tower in each column. Python's zip function is used here, which effectively transposes the grid, grouping all elements at similar column indices together. By iterating over these transposed columns and selecting the maximum value found in each, we compute the zx area.

    zx = sum(max(col) for col in zip(*grid))

After computing the areas individually, we sum them to get the total area of all three projections, which is returned as the final result. The algorithm's complexity is linear in the number of cells in the grid, making it O(n^2), where n is the dimension of the grid. The solution is not only intuitive but also showcases the power and readability of Python's list comprehensions and built-in functions.

No complex patterns or algorithms are necessary for this problem, as it's more about understanding what each projection looks like and selecting the appropriate elements from the grid to count. The solution, therefore, combines basic iteration and aggregation methods to achieve an efficient and understandable result.

class Solution:
    def projectionArea(self, grid: List[List[int]]) -> int:
        xy = sum(v > 0 for row in grid for v in row)
        yz = sum(max(row) for row in grid)
        zx = sum(max(col) for col in zip(*grid))
        return xy + yz + zx

Ready to land your dream job?

Unlock your dream job with a 2-minute evaluator for a personalized learning plan!

Start Evaluator

Example Walkthrough

Let's go through an example to illustrate the solution approach using a 3 x 3 grid as an example:

Suppose we have the following grid representing the heights of the towers of cubes:

grid = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9]
]

We will walk through the solution by calculating the area of each orthogonal projection.

Top-Down View (xy-plane):

First, we look at how many cubes contribute to the top view. Since all cubes are visible from the top, we count all cells with a height greater than zero.

1 2 3
4 5 6
7 8 9

From this, we can see that all 9 cells contribute to the top view (as all heights are non-zero). Thus, the xy area is 9.

Front View (yz-plane):

Now, we look at each row to find the tallest tower, as this would be visible from the front.

1 2 3  <- Tallest in 1st row is 3
4 5 6  <- Tallest in 2nd row is 6
7 8 9  <- Tallest in 3rd row is 9

The heights of the tallest towers are 3, 6, and 9, and the yz area is the sum of these which is 3 + 6 + 9 = 18.

Side View (zx-plane):

For the side view, we need the tallest tower in each column. By looking at each column:

1
4
7 <- Tallest in 1st column is 7

2
5
8 <- Tallest in 2nd column is 8

3
6
9 <- Tallest in 3rd column is 9

The tallest towers from each column have heights 7, 8, and 9. Thus, the zx area is the sum of these which is 7 + 8 + 9 = 24.

Finally, add up all the areas to get the total surface area of all projections:

xy + yz + zx = 9 + 18 + 24 = 51

So according to our solution, the final result for the given 3 x 3 grid is 51.

Solution Implementation

1class Solution:
2    def projectionArea(self, grid: List[List[int]]) -> int:
3        # Calculate the XY projection area by counting all non-zero values in the grid
4        xy_projection = sum(value > 0 for row in grid for value in row)
5      
6        # Calculate the YZ projection area by summing up the tallest building in each row
7        yz_projection = sum(max(row) for row in grid)
8      
9        # Calculate the ZX projection area by summing up the tallest building in each column.
10        # zip(*grid) creates an iterator that aggregates elements from each row into columns.
11        zx_projection = sum(max(column) for column in zip(*grid))
12      
13        # Return the sum of the three projection areas.
14        return xy_projection + yz_projection + zx_projection
15
1class Solution {
2  
3    // Method to calculate projection area of a 3D shape
4    // represented by a grid onto the xy, yz, and zx planes.
5    public int projectionArea(int[][] grid) {
6        int areaXY = 0; // Area projection onto the XY plane
7        int areaYZ = 0; // Area projection onto the YZ plane
8        int areaZX = 0; // Area projection onto the ZX plane
9      
10        // We assume grid is a square (n x n matrix), so we use grid.length to get 'n'.
11        for (int i = 0, n = grid.length; i < n; ++i) {
12            int maxYForYZ = 0; // Max height in y-direction for YZ projection
13            int maxXForZX = 0; // Max height in x-direction for ZX projection
14          
15            // Iterate over each row and column to calculate heights
16            for (int j = 0; j < n; ++j) {
17              
18                // Increment area counter for XY plane if height > 0
19                if (grid[i][j] > 0) {
20                    areaXY++;
21                }
22              
23                // Determine the tallest point in the current row for YZ projection
24                maxYForYZ = Math.max(maxYForYZ, grid[i][j]);
25              
26                // Determine the tallest point in the current column for ZX projection
27                maxXForZX = Math.max(maxXForZX, grid[j][i]);
28            }
29          
30            // Accumulate the maximum height of the grid for YZ and ZX projections
31            areaYZ += maxYForYZ;
32            areaZX += maxXForZX;
33        }
34      
35        // The total projection area is the sum of all three individual areas
36        return areaXY + areaYZ + areaZX;
37    }
38}
39
1#include <vector>
2#include <algorithm>
3using namespace std;
4
5class Solution {
6public:
7    int projectionArea(vector<vector<int>>& grid) {
8        int areaXY = 0; // Initialize the area of the projection onto the XY plane
9        int areaYZ = 0; // Initialize the area of the projection onto the YZ plane
10        int areaZX = 0; // Initialize the area of the projection onto the ZX plane
11
12        // Iterate over the grid to calculate the areas of the projections
13        for (int i = 0, n = grid.size(); i < n; ++i) {
14            int maxRowValue = 0; // The maximum value in the current row for YZ projection
15            int maxColValue = 0; // The maximum value in the current column for ZX projection
16
17            for (int j = 0; j < n; ++j) {
18                // Count the number of cells with a value greater than 0 for the XY projection
19                areaXY += grid[i][j] > 0 ? 1 : 0; 
20
21                // Find the maximum value in the i-th row for YZ projection
22                maxRowValue = max(maxRowValue, grid[i][j]);
23
24                // Find the maximum value in the j-th column for ZX projection
25                maxColValue = max(maxColValue, grid[j][i]);
26            }
27
28            // Add the maximum row value to the YZ projection's area
29            areaYZ += maxRowValue;
30          
31            // Add the maximum column value to the ZX projection's area
32            areaZX += maxColValue;
33        }
34
35        // Return the total projection area by summing up the individual projections
36        return areaXY + areaYZ + areaZX;
37    }
38};
39
1function projectionArea(grid: number[][]): number {
2    // Get the size of the grid.
3    const size = grid.length;
4    // Calculate the top-view projection area, counting the cells that are non-zero.
5    let totalArea = grid.reduce((total, row) => total + row.reduce((count, cell) => count + (cell > 0 ? 1 : 0), 0), 0);
6  
7    // Iterate over each row and each column to find the maximum heights
8    // and calculate the side-view projection areas.
9    for (let i = 0; i < size; i++) {
10        let maxRowHeight = 0; // Max height in the current row.
11        let maxColHeight = 0; // Max height in the current column.
12        for (let j = 0; j < size; j++) {
13            // Find the maximum height of the current row by comparing heights of the cells.
14            maxRowHeight = Math.max(maxRowHeight, grid[i][j]);
15            // Find the maximum height of the current column by comparing heights of the cells.
16            maxColHeight = Math.max(maxColHeight, grid[j][i]);
17        }
18        // Add the maximum heights to the total projection area.
19        totalArea += maxRowHeight + maxColHeight;
20    }
21  
22    // Return the total projection area.
23    return totalArea;
24}
25

Time and Space Complexity

The time complexity of the given code is O(N^2) where N is the number of rows (or columns, assuming the grid is square). This is due to the three separate loops that iterate over all N*N elements of the grid or its transposition. The first loop iterates over all elements for the xy shadow, the second loop iterates over each row to find the maximum for yz shadow, and the third loop iterates over each transposed column to find the maximum for the zx shadow.

The space complexity of the code is O(N) because of the space required for the transposition of the grid (zip(*grid)). This creates a list of tuples where each tuple is a column from the original grid. The space is linear with the number of columns N.

Learn more about how to find time and space complexity quickly using problem constraints.


Discover Your Strengths and Weaknesses: Take Our 2-Minute Quiz to Tailor Your Study Plan:
Question 1 out of 10

Which algorithm should you use to find a node that is close to the root of the tree?


Recommended Readings

Want a Structured Path to Master System Design Too? Don’t Miss This!


Load More