字节青训营最大矩形面积问题

93 阅读2分钟

问题描述

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

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

注意

在后续回看代码时发现 对于第二个遍历为什么时n-j+1具有疑问?

其限制条件在于切片时array[j:i+j],i+j要小于等于列表索引,也就是其最大为n 这里回顾一下python中切片是左闭右开所以i+j取不到n也就是

i+j<=n

就得到了

j<=n-i

又因为range也是左闭右开所以这里应该是

j< n-i+1

具体代码(Python)

def solution(n, array):
    # Edit your code here
    max_area=0
    #k的可以取值
    for i in range(1,n+1):
        #针对k进行滑动窗口
        for j in range(n-i+1):
            min_height=min(array[j:j+i])
            current_area=min_height*i
            if current_area >=max_area:
                max_area=current_area
    return max_area


if __name__ == "__main__":
    # Add your test cases here
    print(solution(5, [1, 2, 3, 4, 5]) == 9)

这道题是滑动窗口的变种的一道题目,不同点在于窗口的大小是不确定的,但是多加一个遍历即可 #k的可以取值 for i in range(1,n+1): 需要注意的是在计算最小高度遍历过程中,窗口是否会越界列表。 最佳的方法就是在窗口大小确定的情况下,限制其有一个安全距离与数组的边界,也就是n-i+1