这题一看就是DP,不过我一开始想的状态是f[i][0/1],表示到第i个位置,第i个灯亮或不亮所需要的最小值,后来看别人写的都只有一维,即f[i]表示到第i个位置的最小值,仔细想想这两种状态是有所区别的,第一种状态的转移比较显然,通过优化后也能AC,这里主要讲一下第二种转移,f[i]=min(f[j])+a[i],其中j的范围是[i-m,i-1],为什么这样转移是对的呢?因为我们相当于默认了第i个灯是亮的,不过这样的话答案就是min{f[i]},i∈[n-m+1,n],我们发现转移的时候相当于是一个滑动窗口,故可以用单调队列优化,当然也可以用线段树,不过就时间和编写的复杂程度来讲,单调队列是很好的办法。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 1000005
int n,m,a[maxn],ans=1e9,f[maxn];
int que[maxn],h=1,t=0;
int main()
{
scanf("%d%d",&n,&m);
for (int i=1;i<=n;i++) scanf("%d",a+i);
f[0]=0;
for (int i=0;i<=n;i++)
{
while (que[h]<i-m&&h<t) h++;
f[i]=f[que[h]]+a[i];
while (f[que[t]]>=f[i]&&t>=h) t--;
que[++t]=i;
}
for (int i=n;i>=n-m+1;i--) ans=min(ans,f[i]);
printf("%d\n",ans);
return 0;
}