201312-3 最大的矩形（Python）

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

6

3 1 6 5 2 3

10

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)