代码有点丑
//当前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;
}