X 国王有一个地宫宝库。是 n x m 个格子的矩阵。每个格子放一件宝贝。每个宝贝贴着价值标签。 地宫的入口在左上角,出口在右下角。
小明被带到地宫的入口,国王要求他只能向右或向下行走。
走过某个格子时,如果那个格子中的宝贝价值比小明手中任意宝贝价值都大,小明就可以拿起它(当然,也可以不拿)。
当小明走到出口时,如果他手中的宝贝恰好是k件,则这些宝贝就可以送给小明。
请你帮小明算一算,在给定的局面下,他有多少种不同的行动方案能获得这k件宝贝。
——
输入格式:
输入一行3个整数,用空格分开:n m k (1<=n,m<=50, 1<=k<=12)
接下来有 n 行数据,每行有 m 个整数 Ci (0<=Ci<=12)代表这个格子上的宝物的价值
输出格式:
要求输出一个整数,表示正好取k个宝贝的行动方案数。该数字可能很大,输出它对 1000000007 取模的结果。
——
样例:
输入:
2 3 2
1 2 3
2 1 5
输出:
14
————————————————————————————————————————————
思路:
创建一个四维数组:H[x][y][z][m] 表示在x,y位置拥有z件宝物,手中宝物的最大价值为m这个状态。
然后进行记忆化递归即可。
★:因为每个状态都可能会有两个子状态:拿,不拿,所以要注意每个位置的状态。
#include <iostream>
#include <cstring>
using namespace std;
#define mod %1000000007
int H[55][55][15][15];//位置,拥有的宝物数量,拥有的最大价值
int pal[55][55];//迷宫
int dir[2][2] = { 1, 0, 0, 1 };//方向
int n, m, k;
//当前的位置,当前的宝物数量,当前宝物数量的最大值
int dfs( int x, int y, int num, int _max ) {
//检查这个点有没有被标记
if( H[x][y][num][_max] != -1 ) return H[x][y][num][_max];
//到达出口的情况
if( x == n&& y == m ) {
if( num == k - 1&& _max < pal[x][y] ) {
return H[x][y][num + 1][pal[x][y]] = 1;
}
else if( num == k ) {
return H[x][y][num][_max] = 1;
}
else return H[x][y][num][_max] = 0;
}
//继续行走
long long int sch = 0;
int lx,ly;
for( int i = 0; i < 2; i++ ) {
lx = x + dir[i][0];
ly = y + dir[i][1];
if( lx <= n&& ly <= m ) {
if( pal[x][y] > _max ) {
sch += dfs( lx, ly, num + 1, pal[x][y] );
}
sch += dfs( lx, ly, num, _max );
}
}
return H[x][y][num][_max] = (sch)mod;
}
int main() {
cin >> n >> m >> k;
for( int i = 1; i <= n; i++ ) {
for( int j = 1; j <= m; j++ ) cin >> pal[i][j];
}
memset( H, -1, sizeof( H ));
cout << dfs( 1, 1, 0, -1 );//_max设置为-1因为价值可能为0
return 0;
}