描述 Description
烽火台又称烽燧,是重要的防御设施,一般建在险要处或交通要道上。一旦有敌情发生,白天燃烧柴草,通过浓烟表达信息:夜晚燃烧干柴,以火光传递军情。在某两座城市之间有n个烽火台,每个烽火台发出信号都有一定的代价。为了使情报准确的传递,在m个烽火台中至少要有一个发出信号。现输入n、m和每个烽火台发出的信号的代价,请计算总共最少需要话费多少代价,才能使敌军来袭之时,情报能在这两座城市之间准确的传递!!!
输入格式 Input Format
第一行有两个数n,m分别表示n个烽火台,在m个烽火台中至少要有一个发出信号。
第二行为n个数,表示每一个烽火台的代价。
输出格式 Output Format
一个数,即最小代价。
样例
5 3
1 2 5 6 2
4
时间限制 Time Limitation
各个测试点1s
注释 Hint
1<=n,m<=1,000,000
分析
要用动态规划的方法解决。我们可以写出这样的方程f[i]:=min{f[j]}+a[i](i-m<=j<i-1)(因为要保证i之前的3个中必须存在被点亮的烽火台)。单纯这样循环会造成超时。
我们想到了用单调队列进行优化,由于随着i的循环,每次只有一个i进入决策区间也只有一个i出决策区间,由于每次选取决策区间中的最小值,所以维护一个单调递增序列,每次取出队首元素即可。
为什么可以将队尾元素无情的删去呢?由于后进队的序列同时满足在原序列中的位置更靠后和其在动态规划中的价值更大。这样选取这个元素就要比选取之前的任何一个决策要优,所以之前被删掉的决策都是无用的。
这道题的本质就是用单调队列维护了决策本身的价值和其在原序列中位置的同时单调。
要特别注意单调队列中的值是决策在原决策序列中的位置。
#include <stdio.h>
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
#define ll long long
const int INF=1<<30;
const int N=1000002;
int h[N],dp[N],q[N],n,k;
/*
dp[0] = 0;
for (i = 1;i <= n;i++) {
for (j = i - 1;j >= 0 && j >= i - k;j--) {
if (dp[j] + h[i-1] < dp[i])
dp[i] = dp[j] + h[i-1];
}
}
*/
int main()
{
int i;
while (~scanf("%d%d",&n,&k))
{
memset(dp,0,sizeof(dp));
int l,r;
l=r=0;
for (i=1;i<=n;i++)
scanf("%d",&h[i]);
for (i=1;i<=n;i++)
{
while (l<=r&&q[l]<i-k)
l++; //这几个的顺序不能发生错乱,因为要从开始找,已找到就记录下来
dp[i]=dp[q[l]]+h[i];
while (l<=r&&h[i]<=h[q[r]])
r--;
q[++r]=i;
}
ll sum=INF;
for (i=n-k+1;i<=n;i++)
if (sum>dp[i])
sum=dp[i];
printf("%lld\n",sum);
}
return 0;
}