1212.地宫取宝 原题链接https://www.acwing.com/problem/content/1214/对于一些基础的二维的动态分析( f[i][j] ),在题目添加了很多限制条件后会变得很复杂,而这时我们可以考虑增加状态方程的纬度,来简化问题。
本题中并不像其他题让你输出走到右下角后的和最大值,而是要求
- 走到右下角后手中要有k件物品,即限制了在走的过程中对考虑对物品捡与不捡
- 捡物品也不是想捡就捡,只要当手中的所有物品的最大价值C都比当前位置的物品的价值小才可以捡
本题要求的是输出走到右下角时,手中刚好有k件物品的方案数。那么毫无疑问,我们可以给状态方程增加一个维度表示走到位置(i,j)时手中有k件物品,即f[i][j][k]
这时我们就解决了第一个限制条件,对于第二个条件,我们一样想怎样表示当走到(i,j)时手中物品的最大价值呢,我们发现这个最大价值与此时手中的物品数也是有关系的,即捡了k件物品,一定会有一个物品的最大值c,这样第四个维度也就出来了
状态方程:,表示在位置(i,j)时,手中有k件物品的最大值c的方案数,那么最终的结果就是把上c的所有取值求和,因为题目问的是手中有k件物品,而最大价值可以是任何值
代码
#include <bits/stdc++.h>
using namespace std;
const int N = 55;
#define p 1000000007
int a[N][N],f[N][N][N][N];
int n,m,K;
int main()
{
cin >> n >> m >> K;
for(int i=1; i<=n; i++)
for(int j=1; j<=m; j++)
{
cin >> a[i][j];
a[i][j]++;
}
f[1][1][1][a[1][1]] = 1;
f[1][1][0][0] = 1;
for(int i=1; i<=n; i++)
for(int j=1; j<=m; j++)
for(int u=0; u<=K; u++)
for(int v=0; v<=13; v++)
{
int &val = f[i][j][u][v];
//不取(i,j)位置上的宝贝
val = (val+f[i-1][j][u][v])%p;
val = (val+f[i][j-1][u][v])%p;
//取 u>0,且此时宝贝的价值要和w[i][j]一样,我们才计算
if( v==a[i][j])
{
for(int c=0; c<v; c++)
{
val = (val+f[i-1][j][u-1][c])%p;
val = (val+f[i][j-1][u-1][c])%p;
}
}
}
int res=0;
for(int i=1; i<=13; i++)
res = (res+f[n][m][K][i])%p;
cout<<res;
return 0;
}