給定一個數組序列, 需要求選出一個區間, 使得該區間是所有區間中經過如下計算的值最大的一個:
區間中的最小數 * 區間所有數的和最後程序輸出經過計算後的最大值即可,不需要輸出具體的區間。
如給定序列 [6 2 1]則根據上述公式, 可得到所有可以選定各個區間的計算值:
[6] = 6 * 6 = 36;
[2] = 2 * 2 = 4;
[1] = 1 * 1 = 1;
[6,2] = 2 * 8 = 16;
[2,1] = 1 * 3 = 3;
[6, 2, 1] = 1 * 9 = 9;
從上述計算可見選定區間 [6] ,計算值爲 36, 則程序輸出爲 36。
區間內的所有數字都在[0, 100]的範圍內;
輸入描述:
第一行輸入數組序列長度n,第二行輸入數組序列。
對於 50%的數據, 1 <= n <= 10000;
對於 100%的數據, 1 <= n <= 500000;
輸出描述:
輸出數組經過計算後的最大值。
示例1
輸入
3
6 2 1
輸出
36
遞增棧,參考leetcode84. 柱狀圖中最大的矩形:
n = int(input())
data = list(map(int, input().split()))
stack = [] # 遞增棧
data.append(0)
res = 0
for i, num in enumerate(data):
while stack and num <= data[stack[-1]]: # 找到最高點
loc = stack.pop(-1)
begin = stack[-1]+1 if stack else 0
res = max(res, data[loc]*sum(data[begin:i]))
stack.append(i) # 記錄索引
print(res)