201312-3 最大的矩形

【题目描述】

在横轴上放了$n$个相邻的矩形，每个矩形的宽度是1，而第$i(1 ≤ i ≤ n)$个矩形的高度是$h_i$。这$n$个矩形构成了一个直方图。例如，下图中六个矩形的高度就分别是3, 1, 6, 5, 2, 3。

请找出能放在给定直方图里面积最大的矩形，它的边要与坐标轴平行。对于上面给出的例子，最大矩形如下图所示的阴影部分，面积是10。

【输入格式】

第一行包含一个整数n，即矩形的数量$(1 ≤ n ≤ 1000)$。

第二行包含$n$ 个整数$h_1, h_2, … ,h_n$，相邻的数之间由空格分隔。$(1 ≤ h_i ≤ 10000)$。$h_i$是第i个矩形的高度。

【输出格式】

输出一行，包含一个整数，即给定直方图内的最大矩形的面积。

【样例1 输入】

6
3 1 6 5 2 3

【样例1 输出】

【解题思路】

把输入数据转化为集合lst_num，防止遍历时重复
遍历lst_num，统计比目前大于或等于num连续值的个数count
和历史最大面积与当前面积num * count比较，更新最大面积

Notice：更新面积时，可能会把当前num下最后一个连续矩形忘掉

【解题程序】

n = int(input())
lst = list(map(int, input().split()))

lst_set = set(lst)
area = 0
for num1 in lst_set:
    count = 0
    for num2 in lst:
        if num2 >= num1:
            count += 1 #统计连续大于num1的矩形个数
        else:
            #最后一个矩形连续，无法更新面积
            if area < num1 * count:
                area = num1 * count
            count = 0 #重新统计连续的矩形个数
            
    #最后一个矩形连续，更新面积
    if area < num1 * count:
        area = num1 * count
print(area)

Eighty Percent