读完题目后就想到了post office的模型,可以这么描述这个问题:在连续的数组里面选m个断点(休息时间),将数组分成很多个小数组,且长度不超过r(连续的听课时间),求分组后的最大值。自然的设定了状态dp[i][j]表示前j分钟睡了i分钟的最优值,显然状态的转移
if(dp[i-1][j-1]!=-1) dp[i][j]=max(dp[i-1][j-1],dp[i][j]); if(dp[i-1][k-1]!=-1) dp[i][j]=max(dp[i][j],dp[i-1][k-1]+sum[j]-sum[k]);k是枚举的所有可行的断点,要注意包括j,这里会有一个新的问题,1000*1000*1000的复杂度,果断tle了,看着这个像四边形优化,但是并不是所有断点可行,所以就会出错,不行之后就转向其他方法,因为它都是从上层转移过来的,看看能否记录上层的最优值,发现可行,因为对于每一个dp[i][j]都只由sum[j]-sum[k]+dp[i-1][k]的最值来的,显然只要保存前k个dp[i-1][k]-sum[k]的最大值即可,那么就开个数组保存就好了,更新的时候要注意可行和不可行,在这里wa了几次,后来就a了。
Run ID | Submit Time | Judge Status | Pro.ID | Exe.Time | Exe.Memory | Code Len. | Language | Author |
4522166 | 2011-08-31 19:11:59 | Accepted | 3905 | 218MS | 8148K | 975 B | C++ | xym2010 |
#include<cstdio> #include<string> #include<cstring> #include<iostream> #include<cmath> #include<algorithm> using namespace std; int dp[1005][1005],sum[1005],a[1005],s[1005][1005]; int main() { int n,m,r; while(scanf("%d%d%d",&n,&m,&r)!=EOF) { for(int i=1;i<=n;i++) { scanf("%d",&a[i]); sum[i]=sum[i-1]+a[i]; } memset(dp,-1,sizeof(dp)); memset(s,-0x1f1f1f1f,sizeof(s)); dp[0][0]=0;s[0][0]=-sum[1]; for(int i=1;i<=n;i++) { if(i>=r) { dp[0][i]=sum[i]; } dp[i][i]=0; s[0][i]=max(s[0][i-1],dp[0][i]-sum[i+1]); } for(int i=1;i<=m;i++) { for(int j=i;j<=n;j++) { if(dp[i-1][j-1]!=-1) dp[i][j]=max(dp[i-1][j-1],dp[i][j]); if(j-r>=i&&s[i-1][j-r-1]!=-0x1f1f1f1f) dp[i][j]=max(dp[i][j],sum[j]+s[i-1][j-r-1]); if(dp[i][j]!=-1) s[i][j]=max(s[i][j-1],dp[i][j]-sum[j+1]); else s[i][j]=s[i][j-1]; } } printf("%d\n",dp[m][n]); } return 0; }