poj 3661 Running

http://poj.org/problem?id=3661

二维DP

#include<iostream>
#include<cstring>
#include<string>
#include<cmath>
#include<cstdio>

using namespace std;

const int N=10005;
const int M=505;
int L[N];
int rest[N][M];
int norest[N][M];
int main()
{
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;++i)
        {
            scanf("%d",&L[i]);
        }
        memset(rest,-1,sizeof(rest));
        memset(norest,-1,sizeof(norest));
        rest[1][0]=0;
        norest[1][1]=L[1];
        for(int i=1;i<n;++i)
        {
            for(int j=0;j<=m;++j)
            {
                if(rest[i][j]!=-1)
                {
                    if(j>0)
                    rest[i+1][j-1]=max(rest[i][j],rest[i+1][j-1]);
                    else
                    {
                        rest[i+1][0]=max(rest[i][j],rest[i+1][0]);
                        norest[i+1][1]=max(norest[i+1][1],L[i+1]+rest[i][j]);
                    }
                }
                if(norest[i][j]!=-1)
                {
                    if(j==m)
                    rest[i+1][m-1]=max(rest[i+1][m-1],norest[i][j]);
                    else
                    {
                        int l=max(j-1,0);
                        rest[i+1][l]=max(rest[i+1][l],norest[i][j]);
                        norest[i+1][j+1]=max(norest[i+1][j+1],L[i+1]+norest[i][j]);
                    }

                }
            }
        }
        int MAX=0;
        MAX=max(rest[n][0],MAX);
        MAX=max(norest[n][0],MAX);
        printf("%d\n",MAX);
    }
    return 0;
}