【蓝桥杯真题】地宫取宝(搜索->记忆化搜索详解)
链接
题目描述
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
思路
这个题目分析题意可以得到是一个dfs递归的题目，因为要搜索出每一种可能的情况。并且这个问题不需要回溯（只是在地图上有两个方向可以选择，并且可以选择是否拾起物体，回溯是不取这个物体，这个物体需要另做他用，这个问题不需要考虑）。
接下来就是如何设计这个dfs。
1.首先考虑递归参数。当前走到的地图的位置（i，j），当前选择出的物品的数目cnt，当前的最大物品的质量max。
2.递归出口是走到了地图的出口，如果当前物品数量等于k，ans++，如果当前的物品数量等于k-1并且最后一个物品的重量大于max，ans++。另外一个递归出口是走出边际和选择的物品数量大于k。
dfs代码如下：
void dfs(int i,int j,int cnt,int max){
if(cnt>k||i>n||j>m)return;
if(i==n&&j==m){
if(cnt==k||(cnt==k-1&&maze[i][j]>max))(++ans)%MOD;
return;
}
dfs(i,j+1,cnt,max);
dfs(i+1,j,cnt,max);
if(maze[i][j]>max){
dfs(i+1,j,cnt+1,maze[i][j]);
dfs(i,j+1,cnt+1,maze[i][j]);
}
}
这样会超时，因为在每一点可以选择的方案是2 or 4，这是一个指数级别搜索量。考虑到到达某一个点会有多种方案并且他们的状态是相同的，所以之前的方案存在重复计算，应该设计记忆型搜索,记忆递归函数应该返回一个值。
代码
// 递归改记忆型递归并不复杂，由于存在重复计算的情况
// 重复计算，就是dfs的参数一样，也就是达到目前的状态可能有许多方式，运用记忆化搜索，对于重复的位置只计算一次。
// 所以用一个标记数组，数组的参数与dfs的参数相同
long long dfs2(int i,int j,int cnt,int max){
//查缓存
if(cache[i][j][cnt][max+1]!=-1)return cache[i][j][cnt][max+1];
long long ans=0;//记得一定是局部变量
if(cnt>k||i>n||j>m)return 0;
if(i==n&&j==m){
if(cnt==k||(cnt==k-1&&maze[i][j]>max))++ans;
return ans;
}
ans+=dfs2(i,j+1,cnt,max);
ans+=dfs2(i+1,j,cnt,max);
if(maze[i][j]>max){
ans+=dfs2(i+1,j,cnt+1,maze[i][j]);
ans+=dfs2(i,j+1,cnt+1,maze[i][j]);
}
// 在return之前一定写缓存
cache[i][j][cnt][max+1]=ans%MOD;
return ans%MOD;
}
完整代码
#include<iostream>
#include<cmath>
#include<sstream>
#include<cstring>
using namespace std;
const int MOD=1000000007;
const int N=55;
int maze[N][N];
int vis[N][N];
int n,m,k;
long long ans=0;
int cache[N][N][13][14];
void dfs(int i,int j,int cnt,int max){
if(cnt>k||i>n||j>m)return;
if(i==n&&j==m){
if(cnt==k||(cnt==k-1&&maze[i][j]>max))(++ans)%MOD;
return;
}
dfs(i,j+1,cnt,max);
dfs(i+1,j,cnt,max);
if(maze[i][j]>max){
dfs(i+1,j,cnt+1,maze[i][j]);
dfs(i,j+1,cnt+1,maze[i][j]);
}
}
// 递归改记忆型递归并不复杂，由于存在重复计算的情况
// 重复计算，就是dfs的参数一样，也就是达到目前的状态可能有许多方式，运用记忆化搜索，对于重复的位置只计算一次。
// 所以用一个标记数组，数组的参数与dfs的参数相同
long long dfs2(int i,int j,int cnt,int max){
//查缓存
if(cache[i][j][cnt][max+1]!=-1)return cache[i][j][cnt][max+1];
long long ans=0;//记得一定是局部变量
if(cnt>k||i>n||j>m)return 0;
if(i==n&&j==m){
if(cnt==k||(cnt==k-1&&maze[i][j]>max))++ans;
return ans;
}
ans+=dfs2(i,j+1,cnt,max);
ans+=dfs2(i+1,j,cnt,max);
if(maze[i][j]>max){
ans+=dfs2(i+1,j,cnt+1,maze[i][j]);
ans+=dfs2(i,j+1,cnt+1,maze[i][j]);
}
// 在return之前一定写缓存
cache[i][j][cnt][max+1]=ans%MOD;
return ans%MOD;
}
int main(){
cin>>n>>m>>k;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
cin>>maze[i][j];
}
memset(cache,-1,sizeof(cache));
// dfs(1,1,0,-1);
cout<<dfs2(1,1,0,-1)<<endl;
//由于max最开始的值为-1，由于数组中不能访问-1.，所以做一个技巧，max+1
return 0;
}
