1. 程式人生 > >[LeetCode] 807. Max increase to keep city skyline(C++)

[LeetCode] 807. Max increase to keep city skyline(C++)

In a 2 dimensional array grid, each value grid[i][j] represents the height of a building located there. We are allowed to increase the height of any number of buildings, by any amount (the amounts can be different for different buildings). Height 0 is considered to be a building as well. 

At the end, the "skyline" when viewed from all four directions of the grid, i.e. top, bottom, left, and right, must be the same as the skyline of the original grid. A city's skyline is the outer contour of the rectangles formed by all the buildings when viewed from a distance. See the following example.

What is the maximum total sum that the height of the buildings can be increased?

Example:
Input: grid = [[3,0,8,4],[2,4,5,7],[9,2,6,3],[0,3,1,0]]
Output: 35
Explanation: 
The grid is:
[ [3, 0, 8, 4], 
  [2, 4, 5, 7],
  [9, 2, 6, 3],
  [0, 3, 1, 0] ]

The skyline viewed from top or bottom is: [9, 4, 8, 7]
The skyline viewed from left or right is: [8, 7, 9, 3]

The grid after increasing the height of buildings without affecting skylines is:

gridNew = [ [8, 4, 8, 7],
            [7, 4, 7, 7],
            [9, 4, 8, 7],
            [3, 3, 3, 3] ]

Notes:

  • 1 < grid.length = grid[0].length <= 50.
  • All heights grid[i][j] are in the range [0, 100].
  • All buildings in grid[i][j] occupy the entire grid cell: that is, they are a 1 x 1 x grid[i][j] rectangular prism.

 

 

思路:主旨即為把數組裡每一個元素換成所在行、所在列兩個最大值之間較小的那一個。先把每一行,每一列的最大值儲存下來,再遍歷判斷即可。

class Solution {
public:
    int maxIncreaseKeepingSkyline(vector<vector<int>>& grid) {
        int n = grid.size();
        vector<int> row(n, 0);
        vector<int> col(n, 0);
        int tempR = 0;
        int tempC = 0;
        
        for (size_t i = 0; i < grid.size(); i++) {
            for (size_t j = 0; j < grid[i].size(); j++) {
                if (row[i] < grid[i][j]) 
                    row[i] = grid[i][j];
                if (col[j] < grid[i][j]) 
                    col[j] = grid[i][j];
            }            
        }
        
        int ret = 0;
        for(size_t i = 0; i < grid.size(); i++) {
            for (size_t j = 0; j < grid[i].size(); j++) {
                int temp = grid[i][j];
                grid[i][j] = row[i] > col[j] ? col[j] : row[i];
                ret += (grid[i][j] - temp);
            }
        }
        
        return ret;
    }
};