poj2228Naptime——环形DP

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

dp[i][j][0/1]表示前i小时中第j小时睡(1)或不睡(0)的最优值；

注意第一个小时，若睡则对最终取结果有要求，即第n个小时必须睡，所以据此DP两遍；

可以开滚动数组，否则容易MLE。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long ans,f[][][];
int n,u[],b;
int main()
{
    scanf("%d%d",&n,&b);
    for(int i=;i<=n;i++)
        scanf("%d",&u[i]);
    memset(f,-,sizeof f);
    f[][][]=;f[][][]=;
    for(int i=;i<=n;i++)
        for(int j=;j<=min(i,b);j++)
        {
            f[i%][j][]=max(f[(i-)%][j][],f[(i-)%][j][]);
            if(j)f[i%][j][]=max(f[(i-)%][j-][],f[(i-)%][j-][]+u[i]);
        }
    for(int j=;j<=b;j++)
        ans=max(ans,max(f[n%][j][],f[n%][j][]));
    memset(f,-,sizeof f);
    f[][][]=u[];
    for(int i=;i<=n;i++)
        for(int j=;j<=min(i,b);j++)
        {
            f[i%][j][]=max(f[(i-)%][j][],f[(i-)%][j][]);
            if(j)f[i%][j][]=max(f[(i-)%][j-][],f[(i-)%][j-][]+u[i]);
        }
    for(int j=;j<=b;j++)
        ans=max(ans,f[n%][j][]);
    printf("%lld",ans);
    return ;
}