bzoj 1737: [Usaco2005 jan]Naptime 午睡时间 （DP）

题目描述

传送门

题解

$f[i][j][0]$表示到第i个一共选了j段，第i个不选的最大价值
$f[i][j][1]$表示到第i个一共选了j段，第i个必选的最大价值
转移方程$f[i][j][0]=max(f[i-1][j][0],f[i-1][j][1])$
$f[i][j][1]=max(f[i-1][j-1][0],f[i-1][j-1][1]+val[i])$
考虑环的问题，我们可以强制第一个选并且不作为入睡阶段，然后枚举后面与之相邻的个数，计算答案。
因为卡内存，所以需要滚动数组。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 4003
using namespace std;
int f[3][N][3],n,m,val[N],sum[N];
int main()
{
    freopen("a.in","r",stdin);
    freopen("my.out","w",stdout);
    scanf("%d%d",&n,&m);
    for (int i=1;i<=n;i++) scanf("%d",&val[i]);
    for (int i=n;i>=1;i--) sum[i]=sum[i+1]+val[i];
    memset(f,128,sizeof(f)); int INF=f[0][0][0];
    f[0][0][0]=0;
    for (int i=1;i<=n;i++) {
     int t=i%2; 
     for (int j=0;j<=min(m,i);j++) f[t][j][0]=f[t][j][1]=INF;
     f[t][0][0]=0;
     for (int j=1;j<=min(m,i);j++) {
      f[t][j][0]=max(f[t][j][0],f[t^1][j][0]);
      f[t][j][0]=max(f[t][j][0],f[t^1][j][1]);
      f[t][j][1]=max(f[t][j][1],f[t^1][j-1][0]);
      f[t][j][1]=max(f[t][j][1],f[t^1][j-1][1]+val[i]);
     }
    }
    int ans=max(f[n%2][m][0],f[n%2][m][1]);
    memset(f,128,sizeof(f));
    f[1][1][1]=val[1]; 
    for (int i=2;i<=n;i++){
         int t=i%2; 
         for (int j=0;j<=min(m,i);j++) f[t][j][0]=f[t][j][1]=INF;
         for (int j=1;j<=min(m,i);j++) {
          f[t][j][0]=max(f[t][j][0],f[t^1][j][0]);
          f[t][j][0]=max(f[t][j][0],f[t^1][j][1]);
          f[t][j][1]=max(f[t][j][1],f[t^1][j-1][0]);
          f[t][j][1]=max(f[t][j][1],f[t^1][j-1][1]+val[i]);
        }
        int t1=n-i+1;
        if (m>=t1) ans=max(ans,f[t][m-t1][0]+sum[i+1]);
    }
    printf("%d\n",ans);
}