poj 2228 Naptime

题目链接：http://poj.org/problem?id=2228

题目思路：定义 dp[N][B][2]：
dp[i][j][0] = { 现在处于第 i 个小时，之前一共休息了 j 个小时，第 i 个小时休息了。到现在为止获得的最大点数 }
dp[i][j][1] = { 现在处于第 i 个小时，之前一共休息了 j 个小时，第 i 个小时没休息。到现在为止获得的最大点数 }
由于是循环数组。所以就分开两种情况：
1. 第 1 个小时没休息
2. 第 1 个小时休息了

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 110
inline int max(int a,int b)
{
	return a>b?a:b;
}
int min(int a,int b)
{
	return a<b?a:b;
}
int dp[2][3900][2];
int v[3900];
int n,m;
int main()
{
    int i,j,k,ans;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++)
        scanf("%d",&v[i]);
    int tmp=0;
    dp[0][1][1]=0;
   // memset(dp[0],0,sizeof(dp[0]));
    for(j=0;j<=m;j++)
    {
        dp[0][j][0]=-inf;
        dp[0][j][1]=-inf;
    }
    dp[0][1][1]=0;
    for(i=2;i<=n;i++)
    {
        for(j=0;j<=m;j++)
        {
            dp[1-tmp][j][0]=-inf;
            dp[1-tmp][j][1]=-inf;
        }
        for(j=0;j<=m;j++)
        {
            dp[1-tmp][j][0]=max(dp[tmp][j][0],dp[tmp][j][1]);
            if(j>0)
                dp[1-tmp][j][1]=max(dp[tmp][j-1][0],dp[tmp][j-1][1]+v[i]);
        }
        tmp=1-tmp;
    }
    ans=max(dp[tmp][m][0],dp[tmp][m][1]+v[1]);
    for(j=0;j<=m;j++)
    {
        dp[0][j][0]=-inf;
        dp[0][j][1]=-inf;
    }
    dp[0][0][0]=0;
    tmp=0;
    for(i=2;i<=n;i++)
    {
        for(j=0;j<=m;j++)
        {
            dp[1-tmp][j][0]=-inf;
            dp[1-tmp][j][1]=-inf;
        }
        for(j=0;j<=m;j++)
        {
            dp[1-tmp][j][0]=max(dp[tmp][j][0],dp[tmp][j][1]);
            if(j>0)
            dp[1-tmp][j][1]=max(dp[tmp][j-1][0],dp[tmp][j-1][1]+v[i]);
        }
        tmp=1-tmp;
    }
    ans=max(ans,dp[tmp][m][0]);
    ans=max(ans,dp[tmp][m][1]);
    printf("%d\n",ans);
}