问题描述
小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)