题目描述
在一个凹槽中放置了 n 层砖块、最上面的一层有n 块砖,从上到下每层依次减少一块砖。每块砖
都有一个分值,敲掉这块砖就能得到相应的分值,如下图所示。
14 15 4 3 23
33 33 76 2
2 13 11
22 23
31
如果你想敲掉第 i 层的第j 块砖的话,若i=1,你可以直接敲掉它;若i>1,则你必须先敲掉第
i-1 层的第j 和第j+1 块砖。
你现在可以敲掉最多 m 块砖,求得分最多能有多少。
输入输出格式
输入格式:输入文件的第一行为两个正整数 n 和m;接下来n 行,描述这n 层砖块上的分值a[i][j],满足
0≤a[i][j]≤100。
对于 100%的数据,满足1≤n≤50,1≤m≤n*(n+1)/2;
输出格式:输出文件仅一行为一个正整数,表示被敲掉砖块的最大价值总和。
输入输出样例
输入样例#1:
4 5
2 2 3 4
8 2 7
2 3
49
输出样例#1:
19
首先看到这道题后无从下手,觉得无论如何设计状态都会有后效性,不能满足动规条件。
考虑将三角形转90度。变成:
4
3 7
2 2 3
2 8 2 49
这一行就可以由前一行的状态转移过来。
设f[i]j[k]表示一定选第i列前j个数,一共选了k个数的最大得分。
因为选一个(i,j)数,一定将第i行的前j个数都取了,所以容易想到前缀和优化。
f[i][j][k]=max(f[i-1][l][k-j]+s[i][j])(j-1=<l<=i-1)
注意i行可以不选,即j从零开始枚举,c党注意判数组是否越界。(这个很不容易像,会坑50分)
考虑优化:开g数组表示f的倒序最大值,就不用枚举l了,少一重循环,会快很多,初始化较繁琐。
未优化:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=55;
int n,m,ans,a[N][N],c[N][N],s[N][N],f[N][N][N*(N+1)/2];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=n-i+1;j++)
scanf("%d",&a[i][j]);
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
c[i][j]=a[j][n-i+1],s[i][j]=s[i][j-1]+c[i][j];
memset(f,-1,sizeof(f));
f[0][0][0]=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
for(int k=j*(j+1)/2;k<=i*(i-1)/2+j;k++)
for(int g=j-1;g<=i-1;g++)
if(f[i-1][g][k-j]>=0)
f[i][j][k]=max(f[i][j][k],f[i-1][g][k-j]+s[i][j]);
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
ans=max(ans,f[i][j][m]);
printf("%d\n",ans);
return 0;
}
优化后:
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
const int N=55;
int n,m,ans,a[N][N],c[N][N],s[N][N],f[N][N][N*(N+1)/2],g[N][N][N*(N+1)/2];
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
for(int j=1;j<=n-i+1;j++)
scanf("%d",&a[i][j]);
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
c[i][j]=a[j][n-i+1],s[i][j]=s[i][j-1]+c[i][j];
memset(g,-1,sizeof(g));
for(int i=0;i<=m;i++)
g[0][0][i]=0;
for(int i=1;i<=n;i++)//行数
for(int j=i;j>=0;j--)//列数
for(int k=j;k<=m;k++)//一共选了k个
if(g[i-1][max(0,j-1)][k-j]>=0)
{
f[i][j][k]=g[i-1][max(j-1,0)][k-j]+s[i][j];
g[i][j][k]=max(g[i][j+1][k],f[i][j][k]);
}
for(int i=1;i<=n;i++)
for(int j=1;j<=i;j++)
ans=max(ans,f[i][j][m]);
printf("%d\n",ans);
return 0;
}