poj 2228 循环dp（睡觉最大值）

最新推荐文章于 2022-07-11 12:07:29 发布

dumeichen

最新推荐文章于 2022-07-11 12:07:29 发布

阅读量385

点赞数

分类专栏：动态规划——杂

本文链接：https://blog.csdn.net/dumeichen/article/details/48469901

版权

动态规划——杂专栏收录该内容

34 篇文章 1 订阅

订阅专栏

题意：将一天分为N小时，每小时都有一个价值w，有一头牛要睡觉，而它的睡觉是连续的，且第一小时不能算价值，即如果你睡了[a,b]，则你获得的收益是w[a+1]+w[a+2]+……+w[b]，而这头牛可以每天多次睡（可以理解成选若干个时间段睡觉），不过每天的睡觉总时间数不能超过B，求能获得的最大总收益。（不过值得注意的是，“1天”并不是从0~N-1，而可以是从任何一个小时开始到n小时之后，即可以从N-1睡到0）

思路：（http://www.tuicool.com/articles/jmuiUzU）先不考虑环：

f[i][j][0]表示前I个小时，睡了j个小时，且第I个小时牛是醒着的

f[i][j][1]表示前I个小时，睡了j个小时，且第I个小时牛是睡着的

那么易得

f[0][0][0]=f[0][1][1]=0,其他f[][][]=-inf

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]+w[i])

ans=max(f[n-1][B][1],f[n-1][B][0])

但是现在有环，容易想到的是把0~N-1再接到N-1的后面，然后枚举从一个位置开始做N长度的DP，那么总时间复杂度是N^3，肯定是要超时的

不过仔细想想本题环的特点是0和N-1都要睡才会存在环！！！！所以环的本质就是0睡没睡，如果没睡那么很明显就是上面那个式子，那么如果睡了呢？

那么显然ans=f[n-1][B][1]，不过前面怎么调整呢

可以f[0][1][1]=w[0]，其他f[][][]=-inf，那么就强制达到后面的状态转移比从f[0][1][1]开始，即保证0必睡

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <queue>
using namespace std;
#define INF 0x3fffffff
#define clr(s,t) memset(s,t,sizeof(s))
#define N 3810
int dp[2][N][2],s[N];
int n,m;
int main(){
    int i,j,a,b,res=0,tmp=-1;
    scanf("%d %d",&n,&m);
    for(i = 1;i<=n;i++)
        scanf("%d",&s[i]);
    dp[0][0][0] = 0;
    dp[0][0][1] = -INF;
    a = 1,b = 0;
    for(i = 1;i<=n;i++){
        a = 1-a;
        b = 1-b;
        dp[b][0][0] = 0;
        dp[b][0][1] = -INF;
        for(j = 1;j<=min(i,m);j++)
            dp[b][j][0] = dp[b][j][1] = -INF;
        for(j = 1;j<=min(i,m);j++){
            if(j<i){
                dp[b][j][0] = max(dp[a][j][0],dp[a][j][1]);
                tmp = dp[b][j][0];
            }
            dp[b][j][1] = max(dp[a][j-1][1]+s[i],dp[a][j-1][0]);
            tmp = dp[b][j][1];
        }
    }
    res = max(dp[b][m][0],dp[b][m][1]);
    dp[0][1][1] = s[1];
    dp[0][0][0] = dp[0][0][1] = dp[0][1][0] = -INF;
    a = 1,b = 0;
    for(i = 2;i<=n;i++){
        a = 1-a;
        b = 1-b;
        dp[b][0][0] = 0;
        dp[b][0][1] = -INF;
        for(j = 1;j<=min(i,m);j++)
            dp[b][j][0] = dp[b][j][1] = -INF;
        for(j = 1;j<=min(i,m);j++){
            if(j<i)
                dp[b][j][0] = max(dp[a][j][0],dp[a][j][1]);
            dp[b][j][1] = max(dp[a][j-1][1]+s[i],dp[a][j-1][0]);
        }
    }
    res = max(res,dp[b][m][1]);
    printf("%d\n",res);
    return 0;
}