最大矩形面积问题 | 豆包MarsCode AI刷题

94 阅读1分钟

问题描述

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

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

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) 的值为这 k 个相邻元素中的最小值乘以 k。现在,小S希望你能帮他找出对于任意 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

思路及代码

我的思路不同于以k为状态,而是以x为段内最小值为观察对象,向左向右查看其所在段,段的长度即为k,kx即为x所能达到的矩形面积。

def solution(n, array):
    # Edit your code here
    size=0
    for i in range(n):
        k=0
        x = array[i] #当前目标
        for j in range(i,-1,-1): #向左搜
            if array[j]>=x:
                k+=1
            else:
                break
        for j in range(i+1,n):
            if array[j]>=x:
                k+=1
            else:
                break
        if k*x>=size:
            size = k*x
    return size


if __name__ == "__main__":
    # Add your test cases here

    print(solution(5, [1, 2, 3, 4, 5]) == 9)