蓝桥杯--------地宫寻宝

【题目描述】

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  取模的结果。

【法1---简单DFS】

#include <iostream>
using namespace std;
int a[55][55];
int n,m,k;
long long ans=0; 
//走到（x，y）处，已经拥有了pre件，其中价值最大的为max 
void dfs(int x,int y,int pre,int max){
	if(pre>k)return ;
	if(x==n-1&&y==m-1){//已经走到了最后一个格子 
		if(pre==k||(pre==k-1&&a[x][y]>max)){//已经拿满了k件，或者拿了k-1件且当前的宝藏可以拿 
			ans++;
		} 
		return ;
	} 
	 if(x<n-1){//允许往下搜索 
	 	if(a[x][y]>max){//比每一个都大 
	 		dfs(x+1,y,pre+1,a[x][y]);//选择拿起，并更新最大价值 
	 	}
	 	dfs(x+1,y,pre,max);//往下走但选择不拿 
	 }
	 if(y<m-1){//可以往右走 
	 	if(a[x][y]>max){
	 		dfs(x,y+1,pre+1,a[x][y]);//拿起并更新最大价值 
	 	}
	 	dfs(x,y+1,pre,max);//不拿 
	 } 
}
int main(){
	cin>>n>>m>>k;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			cin>>a[i][j];
		}
	}
	dfs(0,0,0,-10);
	cout<<"ans="<<ans%1000000007<<endl;
}

以上的代码，在蓝桥杯试题集中得分只能拿到一半，别看n和m的数据量都不大，但是随着n和m的增多，方案数可以说是很大了，这样的dp运行一定会超时，那么该如何解决这个问题呢？

当有重复的子问题时，这个时候可以想到用动归，也能想到用记忆化搜索的方法。

那什么叫重复子问题呢，简单的说，在dfs这个函数中的四个参数是一样子的，你的代码求解了多次，这就叫重复的子问题。最常用的处理手段，我将我每次算好的dfs的值都储存起来，所以这个时候就需要把dfs的类型从void改成int类型。如何存储呢？

我们采用维数和参数个数一致的数组，在这里我们就创建一个用于记录的数组cache，一维即x，二维即y，三维即i，四维即max（关于这些参数的含义，在上一个代码的注解中）,每一次当传入一个（x,y,i,max）的参数时，首先在cache中找（即为cache[x][y][i][max]），如果有，则立即返回。

实际记忆化搜索只需在原本的dp上进行开头和结尾的处理

开头在cache中直接找，结束的时候再把相应的结果存到cache中。

具体代码如下

#include <iostream>
#include <memory.h>
using namespace std;

int a[100][100];
int n,m,k;

int cache[52][52][13][13];
//表示当前走到(x,y)处，且已经有了i件物品，其中最大价值为max这一状态下，
//走到走到终点(n-1,m-1)的可能方案数; 
int dfs(int x,int y,int i,int max){
	
	//先去查缓存！如果缓存中已经有这种情况的可能方案树则直接返回！ 
	if(cache[x][y][i][max+1]!=-1){
		return cache[x][y][i][max+1];//已经做过相应的解法，就无需往下走 
	}
	long long ans=0;
	if(x>=n||y>=m||i>k)return 0;
	if(x==n-1&&y==m-1){//走到了最后一个格子处 
		if(i==k||(i==k-1&&a[x][y]>max)){
			ans++;
			ans%=1000000007;
		}
		return ans;//如果已经走到边界，则ans要么为1要么为0 
	}
	if(x<n){//能继续往下走 
		if(a[x][y]>max){//能拿起当前位置的物品 
			ans+=dfs(x+1,y,i+1,a[x][y]);
		}
		ans+=dfs(x+1,y,i,max);//也可以选择不拿 
	}
	if(y<m){//能继续往右走 
		if(a[x][y]>max){
			ans+=dfs(x,y+1,i+1,a[x][y]);
		}
		ans+=dfs(x,y+1,i,max);
	} 
	//当前位置的ans数来自于四个不同的选择的方案数总和 
	cache[x][y][i][max+1]=ans%1000000007;
	return cache[x][y][i][max+1];
}

int main(){
	cin>>n>>m>>k;
	for(int i=0;i<n;i++){
		for(int j=0;j<m;j++){
			cin>>a[i][j];
		}
	} 
	memset(cache,-1,sizeof(cache));//全部初始化为-1 
	cout<<dfs(0,0,0,-1)<<endl;
	return 0;
}