图灵杯 爬楼梯

题目描述

由于第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 1005using 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;}