题目描述
由于第m个台阶上有好吃的薯条,所以薯片现在要爬一段m阶的楼梯.
薯片每步最多能爬k个阶梯,但是每到了第i个台阶,薯片身上的糖果都会掉落ai个,现在问你薯片至少得掉多少糖果才能得到薯条?
输入
多组数据输入,每组数据第一行输入两个数字m(1<m<=1000),k(1<=k<m),接下来有m行,每一行输入ai(0<ai<=1000),表示第i个台阶会掉落的糖果.
输出
对于每组数据,输出至少要牺牲的糖果数.
样例输入
5 2
1 2 3 4 5
6 2
6 5 4 3 2 1
样例输出
9
由于第m个台阶上有好吃的薯条,所以薯片现在要爬一段m阶的楼梯.
薯片每步最多能爬k个阶梯,但是每到了第i个台阶,薯片身上的糖果都会掉落ai个,现在问你薯片至少得掉多少糖果才能得到薯条?
输入
多组数据输入,每组数据第一行输入两个数字m(1<m<=1000),k(1<=k<m),接下来有m行,每一行输入ai(0<ai<=1000),表示第i个台阶会掉落的糖果.
输出
对于每组数据,输出至少要牺牲的糖果数.
样例输入
5 2
1 2 3 4 5
6 2
6 5 4 3 2 1
样例输出
9
9
思路:dp公式: d[i]= min(d[i-m] ,d[i-m+1] ,.....d[i-1]) +a[i] .用单调队列维护就不会超时了。
#include <iostream>
#include<stdio.h>
#include<string.h>
#define siz 1005
using namespace std;
int n,k,q[siz],a[siz],d[siz];
int main()
{
int f,b;
while(scanf("%d %d",&n,&k)!=EOF)
{
memset(d,0,sizeof(d));
memset(q,0,sizeof(q));
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
q[1]=0;
f=b=0;
d[0]=0;
for(int i=1;i<=n;i++)
{
int u;
u=q[b];
d[i]=d[q[f]]+a[i];
while(d[u]>d[i]&&f<=b){
b--;
u=q[b];
}
q[++b]=i;
if(q[f]<i-k+1) f++;
}
cout<<d[n]<<endl;
}
return 0;
}