黑书1.5.1例题5 积木游戏
刚开始想按黑书那样去写,四层状态,倒着推。然后,2h过去了,爆炸。从网上搜了搜,等等,你们怎么都就三层状态?认真看一下,woc,我一定是傻疯了写四层。然后瞬间ac。看到一个写四层的,基本都是记忆化搜索过了。我倒推好弱。。推不明白。还是正推好一些。。(网上神犇纷纷表示,这是一道dp水题,好吧orz)说正解,基本都写注释里了:)状态
O(mn)
,决策
O(n)
,状态转移
O(1)
,总时间复杂度
O(n2m)
(黑书上说的正推决策不唯一是这个意思么。。可是倒推蒟蒻真心不行啊/(ㄒoㄒ)/~~)
#include <cstdio>
#include <cstring>
#define N 110
int n,m,dp[N][N][3],ans=-1;//dp[i][j][k]表示第i堆顶面是第j块的k面时获得的最高高度
struct node{
int len[3];
}a[N];
inline int read(){
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
return x*f;
}
inline int max(int x,int y){return x>y?x:y;}
inline bool ok(int i1,int k1,int i2,int k2){
int x=a[i1].len[(k1+3-1)%3],y=a[i1].len[(k1+1)%3];
int xx=a[i2].len[(k2+3-1)%3],yy=a[i2].len[(k2+1)%3];
if(x>=xx&&y>=yy||x>=yy&&y>=xx) return 1;
return 0;
}
int main(){
freopen("buildinggame.in","r",stdin);
freopen("buildinggame.out","w",stdout);
n=read();m=read();
for(int i=1;i<=n;++i) a[i].len[0]=read(),a[i].len[1]=read(),a[i].len[2]=read();
for(int i=1;i<=m;++i)//现在在搭第i堆
for(int j=i;j<=n;++j)//用的是第j块。
for(int k=0;k<3;++k)//用的是第j块的k面作为下表面
for(int jj=i-1;jj<j;++jj)//摆第j块之前的顶面是jj块。
for(int kk=0;kk<3;++kk){//jj块的上表面是kk面
dp[i][j][k]=max(dp[i][j][k],dp[i-1][jj][kk]+a[j].len[k]);//jj块是i-1堆的顶面,j块作为第i堆的第一块
if(ok(jj,kk,j,k)) dp[i][j][k]=max(dp[i][j][k],dp[i][jj][kk]+a[j].len[k]);//jj块是第i堆的顶面,j块摆在他上面
}
for(int i=m;i<=n;++i)
for(int k=0;k<3;++k) ans=max(ans,dp[m][i][k]);
printf("%d\n",ans);
return 0;
}