AcWing 1089. 烽火传递 Python3 单调队列优化DP
原题链接
中等
作者:
皓首不倦
,
2020-07-29 18:26:42
,
所有人可见
,
阅读 510
from collections import deque
n, m = map(int, input().split())
cost = list( map(int, input().split()) )
# dp(i) 表示前i个烽火台进行选择,且最后一个烽火台点燃的所有方案中,最小开销的数值
# dp(i) = cost(i) + min(dp(i-1), dp(i-2), ... dp(i-m)) 其实dp(i)就依赖于前面长度为m
# 滑动窗口中的最小值
dp = [0] * n
que = deque()
for idx, val in enumerate(cost):
if idx < m:
dp[idx] = cost[idx]
else:
dp[idx] = que[0][1] + cost[idx]
while len(que) > 0 and idx - que[0][0] >= m:
que.popleft()
while len(que) > 0 and dp[idx] <= que[-1][1]:
que.pop()
que.append( (idx, dp[idx]) )
print( min(dp[-m:]) )