Acwing 288.休息时间 (环形DP)

题目

在某个星球上,一天由 N 个小时构成,我们称0点到1点为第1个小时、1点到2点为第2个小时,以此类推。

在第 i 个小时睡觉能够恢复Ui点体力。

在这个星球上住着一头牛,它每天要休息B个小时。

它休息的这B个小时不一定连续,可以分成若干段,但是在每段的第一个小时,它需要从清醒逐渐入睡,不能恢复体力,从下一个小时开始才能睡着。

为了身体健康,这头牛希望遵循生物钟,每天采用相同的睡觉计划。

另外,因为时间是连续的,即每一天的第N个小时和下一天的第1个小时是相连的(N点等于0点),这头牛只需要在每N个小时内休息够B个小时就可以了。

请你帮忙给这头牛安排一个睡觉计划,使它每天恢复的体力最多。

输入格式
第1行输入两个空格隔开的整数N和B。

第2..N+1行,第 i+1 行包含一个整数Ui。

输出格式
输出一个整数,表示恢复的体力值。

数据范围
3≤N≤3830
2≤B<N
0≤Ui≤200000
输入样例:
5 3
2
0
3
1
4
输出样例:
6
样例解释
这头牛每天3点入睡,睡到次日1点,即[1,4,2]时间段休息,每天恢复体力值最大,为0+4+2=6。

思路

注意,当他说只需要满足n小时里面的b小时即可,意思就是在一个环形的时间圈内选择,那么这个时候我们就考虑两种思路,一种是破环成链,第二种是对边界进行分类讨论。那么我们这里采用第二种的方法,第一种方法会爆内存。所以我们接下来考虑dp的状态设计,思考一下,某个小时可以选择睡或者不睡,然后我们有n个个小时,在里面选择b个,所以DP[N][B][1]和DP[N][B][0]两者里面的最大值就是答案,状态转移方程的话自己推一下啦,分类就可以。

代码实现

#include<cstdio>
#include<algorithm>
#include<vector>
#include<queue>
#include<map>
#include<iostream>
#include<cstring>
#include<cmath>
using namespace std;
#define rep(i,f_start,f_end) for (int i=f_start;i<=f_end;++i)
#define per(i,n,a) for (int i=n;i>=a;i--)
#define MT(x,i) memset(x,i,sizeof(x) )
#define rev(i,start,end) for (int i=0;i<end;i++)
#define inf 0x3f3f3f3f
#define mp(x,y) make_pair(x,y)
#define lowbit(x) (x&-x)
#define MOD 1000000007
#define exp 1e-8
#define N 1000005 
#define fi first 
#define se second
#define pb push_back
typedef long long ll;
typedef pair<int ,int> PII;
ll gcd (ll a,ll b) {return b?gcd (b,a%b):a; }
inline int read() {
    char ch=getchar(); int x=0, f=1;
    while(ch<'0'||ch>'9') {
        if(ch=='-') f = -1;
        ch=getchar();
    } 
    while('0'<=ch&&ch<='9') {
        x=x*10+ch-'0';
        ch=getchar();
    }   return x*f;
}

const int maxn=4000;
int f[2][maxn][2];
int n,m;
int w[maxn];


int main () {
   freopen ("data.in","r",stdin);
    cin>>n>>m;
    rep (i,1,n) {
        scanf ("%d",&w[i]);
    }   
    MT (f,-0x3f);
    f[1][0][0]=f[1][1][1]=0;
    rep (i,2,n) 
     rep (j,0,m) {
       f[i&1][j][0]=max (f[i-1&1][j][0],f[i-1&1][j][1]);
       f[i&1][j][1]=-inf;
       if (j) f[i&1][j][1]=max (f[i-1&1][j-1][0],f[i-1&1][j-1][1]+w[i]);
     }
    int ans=f[n&1][m][0];
     
    MT (f,-0x3f);
    f[1][0][0]=0,f[1][1][1]=w[1];
    rep (i,2,n) 
     rep (j,0,m) {
         f[i&1][j][0]=max (f[i-1&1][j][1],f[i-1&1][j][0]);
         f[i&1][j][1]=-inf;
         if (j)  f[i&1][j][1]=max (f[i-1&1][j-1][1]+w[i],f[i-1&1][j-1][0]); 
     }

     ans=max (ans,f[n&1][m][1]);
     cout<<ans<<endl;
    fclose (stdin);
   return 0;
}

posted @ 2020-08-07 16:41  Luglucky  阅读(109)  评论(0编辑  收藏  举报