編程題:區間中的最小數 * 區間所有數的和最大

給定一個數組序列, 需要求選出一個區間, 使得該區間是所有區間中經過如下計算的值最大的一個:
區間中的最小數 * 區間所有數的和最後程序輸出經過計算後的最大值即可,不需要輸出具體的區間。

如給定序列 [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)
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章