问题 1436: [蓝桥杯][2014年第五届真题]地宫取宝

题目描述

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

 

原题链接

 

WA和超时了一个中午，总结一下记忆化搜索，其实思路已经到位了，就是学到了很多边界的处理，先从开空间说起，有多少个参数就开多少维空间，memo数组的初始值设为-1，而不是0，不然记忆搜索会退化导致超时，另外一个要注意的就是数组的值范围为0-12，所以你穿入的max不能为0，应该为-1，但是开的空间下标总不能为负数把，所以可以让所有数组里面的值+1就等价于max改为-1了。

 

 

import java.util.Scanner;

public class 地宫取宝 {
	
	static int n,m,k;
	static int[][] a = new int[55][55];	
	static int[][][][] memo = new int[55][55][13][13];
	static int mod = 1000000007;
	
	public static void main(String[] args) {
		Scanner in = new Scanner(System.in);
		n = in.nextInt();
		m = in.nextInt();
		k = in.nextInt();
		for(int i=1;i<=n;i++)
			for(int j=1;j<=m;j++)
				a[i][j] = 1+in.nextInt();
		
		for(int i=0;i<=54;i++)
			for(int j=0;j<=54;j++)
				for(int ii=0;ii<=12;ii++)
					for(int jj=0;jj<=12;jj++)
					memo[i][j][ii][jj]=-1;

		System.out.println(dfs(1,1,0,0));
	}
	
	static int dfs(int x,int y,int t,int max) {
		if(memo[x][y][t][max]!=-1)
			return memo[x][y][t][max];
		memo[x][y][t][max]=0;
		if(x==n && y==m && t==k)
			memo[x][y][t][max] = 1;		

		if(a[x][y]>max && t<k) {
			memo[x][y][t][max] += dfs(x,y,t+1,a[x][y]);	
			memo[x][y][t][max]%=mod;
		}
		
		if(x<n) {
			memo[x][y][t][max] += dfs(x+1,y,t,max);	
			memo[x][y][t][max]%=mod;

		}
		if(y<m) {
			memo[x][y][t][max] += dfs(x,y+1,t,max);
			memo[x][y][t][max]%=mod;
		}
	
		return memo[x][y][t][max];
	}

}

 

 

这几个小时花的挺值的，对记忆的理解更进一步了