图灵杯 爬楼梯

题目描述

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