noi97 ecnuoj 1244 积木游戏(dp,黑书dp例五)

noi97 ecnuoj 1244 积木游戏

代码有点丑

//当前i根柱子，用到前num个，最顶上为b积木，b的k面朝上

int dp[p][num][b][3];

三种决策

1:当前块放在前一堆上，前提是当前块能放上去，dp[p][num+1][num+1][k]

2:当前快新建一堆，dp[p+1][num+1][num+1][k]

3:略过当前块,dp[p][num+1][b][k]

#include<stdio.h>
#include<string.h>
#define MAXN 105
struct block
{
    int len[3];
}aa[MAXN];
int n,m;
//当前i根柱子，用到前num个，最顶上为b积木，b的k面朝上
int dp[MAXN][MAXN][MAXN][3];
int max(int a,int b) {return a>b?a:b;}
int judge(int a,int k1,int b,int k2)
{
    int ea1=aa[a].len[k1],ea2=aa[a].len[(k1+1)%3],
    eb1=aa[b].len[k2],eb2=aa[b].len[(k2+1)%3];
    if((ea1>=eb1&&ea2>=eb2)||(ea1>=eb2&&ea2>=eb1))
        return aa[b].len[(k2+2)%3];
    return 0;
}
int d(int p,int num,int b,int k)
{
    if(dp[p][num][b][k]) return dp[p][num][b][k];
    if(p==m&&num==n) return 0;
    if(p!=m&&num==n) return -1;
    int ans=0;
    //加在前一堆
    if(p)
        for(int i=0;i<3;i++)
        {
            int h=judge(b,k,num+1,i);
            if(h) ans=max(ans,d(p,num+1,num+1,i)+h);
        }
    //新建一堆
    if(p<m)
        for(int i=0;i<3;i++)
        {
            ans=max(ans,d(p+1,num+1,num+1,i)+aa[num+1].len[(i+2)%3]);
        }
    //略过当前块
    ans=max(ans,d(p,num+1,b,k));
    
    dp[p][num][b][k]=ans;
    return ans;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(int i=1;i<=n;i++)
            scanf("%d%d%d",&aa[i].len[0],&aa[i].len[1],&aa[i].len[2]);
        memset(dp,0,sizeof(dp));
        printf("%d\n",d(0,0,0,0));
    }
    return 0;
}