201312-3 最大的矩形（Python）

https://blog.aksy.space/CCF-CSP/201312-3.html

【题目描述】

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

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

【输入格式】

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

第二行包含个整数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)