标题:地宫取宝
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 2 2
1 2
2 1
程序应该输出:
2
再例如,输入:
2 3 2
1 2 3
2 1 5
程序应该输出:
14
资源约定:
峰值内存消耗(含虚拟机) < 256M
CPU消耗 < 2000ms
解题思路:若直接用dfs遍历所有情况,会导致超时,不仅仅是数据量大,还有dfs过程中有处理大量的重复情况。
所以可以采取记忆化搜索,将某一位置(x, y),手上有的宝物数量n,手上宝物中最大的价值mv,记录下来。
dp[x][y][n][mv] 代表该状态下走到终点满足情况的方案数,若第2次,第3次,第。。次再遇到这种情况直接加上dp[][][][]。
#include<iostream>
#include<cstring>
using namespace std;
const int Mod = 1000000007;
long long n, m, goal, maze[55][55];
long long dp[55][55][15][15];
long long find_ways(int x, int y, int treasure, int maxValue){
if(dp[x][y][treasure][maxValue+1] != -1){ //当前状态已经走过(已知该状态到达目标态有几条路径)
return dp[x][y][treasure][maxValue+1];
}
if(x==n && y==m){ //到达终点
if(treasure==goal || (treasure==goal-1 && maze[x][y]>maxValue)){ //若满足要求,记录该状态到达目标态有一条路径
return dp[x][y][treasure][maxValue+1] = 1;
}
return dp[x][y][treasure][maxValue+1] = 0;
}
long long tmp = 0;
if(x+1 <= n){
if(maze[x][y] > maxValue){
tmp = (tmp + find_ways(x+1, y, treasure+1, maze[x][y])%Mod)%Mod;
}
tmp = (tmp + find_ways(x+1, y, treasure, maxValue)%Mod)%Mod;
}
if(y+1 <= m){
if(maze[x][y] > maxValue){
tmp = (tmp + find_ways(x, y+1, treasure+1, maze[x][y])%Mod)%Mod;
}
tmp = (tmp + find_ways(x, y+1, treasure, maxValue)%Mod)%Mod;
}
return dp[x][y][treasure][maxValue+1] = tmp;
}
int main()
{
memset(dp, -1, sizeof(dp));
cin >> n >> m >> goal;
for(int i=1; i<=n; ++i){
for(int j=1; j<=m; ++j){
cin >> maze[i][j];
}
}
cout << find_ways(1, 1, 0, -1);
return 0;
}