青训营-最大矩形面积问题

215 阅读1分钟

问题描述

小S最近在分析一个数组 h1,h2,...,hNh1​,h2​,...,hN​,数组的每个元素代表某种高度。小S对这些高度感兴趣的是,当我们选取任意 kk 个相邻元素时,如何计算它们所能形成的最大矩形面积。

对于 kk 个相邻的元素,我们定义其矩形的最大面积为:

R(k)=k×min(h[i],h[i+1],...,h[i+k−1])R(k)=k×min(h[i],h[i+1],...,h[i+k−1])

即,R(k)R(k) 的值为这 kk 个相邻元素中的最小值乘以 kk。现在,小S希望你能帮他找出对于任意 kk,R(k)R(k) 的最大值。


测试样例

样例1:

输入:n = 5, array = [1, 2, 3, 4, 5]
输出:9

样例2:

输入:n = 6, array = [5, 4, 3, 2, 1, 6]
输出:9

样例3:

输入:n = 4, array = [4, 4, 4, 4]
输出:16

算法实现

#include <iostream>

int solution(int n, int array[]) {
    // Edit your code here
 int max_area = 0;
    
    // 枚举所有可能的起始位置
    for (int i = 0; i < n; i++) {
        int min_height = array[i]; // 记录当前的最小高度
        
        // 枚举每一个结束位置
        for (int j = i; j < n; j++) {
            // 更新最小高度
            min_height = std::min(min_height, array[j]);
            // 计算当前矩形面积
            int current_area = (j - i + 1) * min_height;
            // 更新最大面积
            max_area = std::max(max_area, current_area);
        }
    }
    
    return max_area;
}

int main() {
    // Add your test cases here
    
    std::cout << (solution(5, new int[5]{1, 2, 3, 4, 5}) == 9) << std::endl;
    return 0;
}