致谢:AcWing 1212. 地宫取宝(普通 dp,重点理解)
1、思路怎么想?
(1)同理想框架,动态规划,状态表示和状态计算,
(2)状态表示:这个就很难了,是四维,
1)为什么是四维?
因为里面有坐标i,j从两个方向来,还有k件物品需要维护,另外还有最后一件物品的价值c需要维护,因为需要c来比较,是否取不取,这样就可以表示出,这个问题了。
类似于维护子序列和摘花生的结合。
为什么不能用二维,因为无法维护变量k这个东西,
为什么不用三维,因为无法维护变量c这个东西,这样就不能判断取还是不取,
所以综上,若在有几个变量,就是几维。
2)集合和属性怎样表示?
属性:数量
f[i , j ,k ,c]表示:所有从起点走到( i , j ),并已经取了k件物品,并且最后一件物品的价值为c的方案数,值是数量。
答案就是:f[n , m , k , i ] , i = 1.....13
表示:从起点走到(n,m),且已经取了k件物品,且最后一件物品的价值是1或2或3...13 的所有方案的求和
(3)状态计算
2、代码怎么写?
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
//1、定义
const int N = 55 , mod = 1000000007;
int n ,m ,k;
int w[N][N];
int f[N][N][13][14];//第三位为能得到宝藏的个数k,第四位为宝藏的价值
int main()
{
//2、输入
scanf("%d%d%d" , &n , &m , &k);
for( int i = 1 ; i <= n ; i++ )
for( int j = 1 ; j <= m ; j++ )
{
scanf("%d" , &w[i][j]);
w[i][j] ++;//为了处理边界,价值从1开始,0作为边界值
}
//3、初始化初始位置的初值,即起点,在属性是数量时,要留意初始化顶点状态
f[1][1][1][w[1][1]] = 1; //取左上角
f[1][1][0][0] = 1;//不取左上角
//4、dp过程
for( int i = 1 ; i <= n ; i++ )
for( int j = 1 ; j <= m ; j++ )
{
//1.左上角已经计算(初始化)过了
if(i == 1 && j == 1) continue;
//2.枚举所有方案数,即变量k的取值
//枚举所有的最后取的(最大的)价值
for( int u = 0 ; u <= k ; u++ )
for( int v = 0 ; v <= 13 ; v++ )
{
int &val = f[i][j][u][v];//把f[i][j][u][v]的地址给val,因为后面要多次用到 f[i][j][u][v],写成val更加方便
//3.不取的两种情况
val = (val + f[i-1][j][u][v]) % mod;
val = (val + f[i][j-1][u][v]) % mod;
//4.取的两种情况
//需要满足两条件,k大于0和w[i][j] == v
if(u > 0 && w[i][j] == v)
{
//5.枚举上一步的,所有的,价值小于当前宝藏价值v的,所有宝藏,实则是一个都是方案数的集合
for( int c = 0 ; c < v ; c++ )
{
val = (val + f[i-1][j][u-1][c]) % mod;//左边来的
val = (val + f[i][j-1][u-1][c]) % mod;//右边来的
}
}
}
}
//5、枚举从起点走到(n,m),且已经取了k件物品,且最后一件物品的价值是1或2或3...13 的所有方案的求和
int res = 0;
for( int i = 1 ; i <= 13 ; i++ ) res = (res + f[n][m][k][i]) % mod;//应该注意,涉及mod,一般最后答案仍需再模
printf("%d", res);
return 0;
}
几个问题
(1)为什么对所有的价值加1?
所有有关属性是数量的,需要先初始化起点,
而在本题中初始化左上角起点时,会发生数组下标越界,即在减价值的时候会发生-1下标现象,
所以给每一个价值加1,使其边界加1,从而不会发生越界现象。
孙少安、孙少平是挣扎在贫困线上的青年人,但他们自强不息,依靠自己的顽强毅力与命运抗争,追求自我的道德完善。
---《平凡的世界》鉴赏