POJ2228 Naptime

注意当转移有一部分不合法时特判而不是一起去掉

#include<bits/stdc++.h>
using namespace std;
const int N=3837;
int f[2][N][2],n,a[N],b,ans;
int main(){
        cin>>n>>b;
        for(int i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        memset(f,0xcf,sizeof(f));
        f[1][0][0]=f[1][1][1]=0;
        for(int i=2;i<=n;i++)
            for(int j=0;j<=b;j++){//不可以直接从1开始循环
                f[i&1][j][0]=max(f[(i-1)&1][j][0],f[(i-1)&1][j][1]);
                if(j>=1)f[i&1][j][1]=max(f[(i-1)&1][j-1][0],f[(i-1)&1][j-1][1]+a[i]);
            }
        ans=max(f[n&1][b][0],f[n&1][b][1]);
        memset(f,0xcf,sizeof(f));
        f[1][1][1]=a[1];
        for(int i=2;i<=n;i++)
            for(int j=1;j<=b;j++){
                f[i&1][j][0]=max(f[(i-1)&1][j][0],f[(i-1)&1][j][1]);
                f[i&1][j][1]=max(f[(i-1)&1][j-1][0],f[(i-1)&1][j-1][1]+a[i]);
            }
        ans=max(ans,f[n&1][b][1]);
        cout<<ans<<endl;
    return 0;
}