poj 2228 Naptime dp

这个题目的状态还是比较好想的，dp[i][j]表示已经睡了i个时段，最后睡在j时段的最优值，但是需要处理环的情况，我的做法是算两次，第一次不处理环，第二次强制性要求第一个时段需要睡，然后查看dp[m][n]+a[1]的值是否更优。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int maxn=4e3+9;
int a[maxn],dp[maxn][maxn];

inline int max(int a,int b)
{
    if(a>b) return a;
    else return b;
}

int main()
{
//    freopen("in.txt","r",stdin);
    int n,m;
    while(scanf("%d %d",&n,&m)!=EOF)
    {

        for(int i=1;i<=m;i++)
        for(int j=1;j<=n;j++)
        dp[i][j]=0;
        for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
        for(int k=2,ret;k<=m;k++)
        {
            ret=0;
            for(int i=k;i<=n;i++)
            {
                dp[k][i]=max(ret,dp[k-1][i-1]+a[i]);
                ret=max(ret,dp[k-1][i-1]);
            }
        }
        int ans=0;
        for(int i=m;i<=n;i++)
        ans=max(dp[m][i],ans);

        for(int i=1;i<=m;i++)
        for(int j=1;j<=n;j++)
        dp[i][j]=0;

        dp[2][2]=a[2];
        for(int k=3,ret;k<=m;k++)
        {
            ret=0;
            for(int i=k;i<=n;i++)
            {
                dp[k][i]=max(ret,dp[k-1][i-1]+a[i]);
                ret=max(ret,dp[k-1][i-1]);
            }
        }
        if(m>=2)
        ans=max(ans,dp[m][n]+a[1]);
        cout<<ans<<endl;
    }
    return 0;
}