pku 2228 Naptime 简单DP,写的很挫,卡常数卡过去了。。。

题意大概描述一下
有一只牛,在全天N小时内花B小时睡觉,然后每一个时间点都有一个休息值,假设选择在区间[s,e]内休息,则第s个时间点不能够获得休息值。现在请安排这只牛的作息时间,使得他的休息值最大。
状态dp[i][k][2],i表示前i个小时,k表示休息了k个小时,最后一个状态表示最后一个小时是否在休息。然后由于可以形成环(即第N个小时在休息,则第1个小时能够获得能量),需要2个DP。状态转移应该很简单吧~具体看程序。话说这题谁有nlogn的方法?感觉n2好悬,常数还不小。。本机上跑了近1秒,用位运算、滚动数组等还有800ms,提交上去只有200ms。。看来poj的服务器很NB
贴代码
 1 # include <cstdio>
 2 # include <cstring>
 3 using namespace std;
 4 # define max(a,b) ((a)>(b)?(a):(b))
 5 int data[3850];
 6  int dp[2][2][3850],dp1[2][2][3850];
 7 int main()
 8 {
 9    // freopen("input.txt","r",stdin);
10   //  freopen("output.txt","w",stdout);
11     int n,k;
12     scanf("%d%d",&n,&k);
13     for(int i=0;i<n;i++)
14         scanf("%d",data+i);
15     memset(dp,-1,sizeof(dp));
16     memset(dp1,-1,sizeof(dp1));
17     dp[0][0][0]=0;
18     dp[0][1][1]=0;
19     dp1[0][0][0]=0;
20     dp1[0][1][1]=data[0];
21     int res=0;
22     for(int i=1;i<n;i++)
23     {
24         memset(dp[i%2],-1,sizeof(dp[i%2]));
25         memset(dp1[i%2],-1,sizeof(dp1[i%2]));
26         dp[i%2][0][0]=0;
27         for(int j=1;j<=k;j++)
28         {
29             if(dp[(i-1)%2][0][j]!=-1)
30                 dp[i%2][0][j]=max(dp[i%2][0][j],dp[(i-1)%2][0][j]);
31             if(dp[(i-1)%2][1][j]!=-1)
32                 dp[i%2][0][j]=max(dp[i%2][0][j],dp[(i-1)%2][1][j]);
33             if(dp[(i-1)%2][0][j-1]!=-1)
34                 dp[i%2][1][j]=max(dp[i%2][1][j],dp[(i-1)%2][0][j-1]);
35             if(dp[(i-1)%2][1][j-1]!=-1)
36                 dp[i%2][1][j]=max(dp[i%2][1][j],dp[(i-1)%2][1][j-1]+data[i]);
37             if(dp1[(i-1)%2][0][j]!=-1)
38                 dp1[i%2][0][j]=max(dp1[i%2][0][j],dp1[(i-1)%2][0][j]);
39             if(dp1[(i-1)%2][1][j]!=-1)
40                 dp1[i%2][0][j]=max(dp1[i%2][0][j],dp1[(i-1)%2][1][j]);
41             if(dp1[(i-1)%2][0][j-1]!=-1)
42                 dp1[i%2][1][j]=max(dp1[i%2][1][j],dp1[(i-1)%2][0][j-1]);
43             if(dp1[(i-1)%2][1][j-1]!=-1)
44                 dp1[i%2][1][j]=max(dp1[i%2][1][j],dp1[(i-1)%2][1][j-1]+data[i]);
45         }
46     }
47     res=max(max(res,dp1[(n-1)%2][1][k]),max(dp[(n-1)%2][1][k],dp[(n-1)%2][0][k]));
48     printf("%d\n",res);
49     return 0;
50 }
51 


posted on 2010-11-07 02:48 yzhw 阅读(355) 评论(0)  编辑 收藏 引用 所属分类: DP


只有注册用户登录后才能发表评论。
网站导航: 博客园   IT新闻   BlogJava   知识库   博问   管理


<2010年10月>
262728293012
3456789
10111213141516
17181920212223
24252627282930
31123456

导航

统计

公告

统计系统

留言簿(1)

随笔分类(227)

文章分类(2)

OJ

最新随笔

搜索

积分与排名

最新评论

阅读排行榜