记忆型dfs

标题：地宫取宝 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 请严格按要求输出，不要画蛇添足地打印类似：“请您输入…” 的多余内容。 所有代码放在同一个源文件中，调试通过后，拷贝提交该源码。 注意：不要使用package语句。不要使用jdk1.7及以上版本的特性。 注意：主类的名字必须是：Main，否则按无效代码处理。

一、下面是不加记忆功能的dfs

import java.util.Scanner; public class dfs_地宫取宝 { static int[][] data; private static int n; private static int m; private static int k; public static void main(String[] args) { Scanner in = new Scanner(System.in); n = in.nextInt(); m = in.nextInt(); k = in.nextInt(); data = new int[n][m]; for(int i=0;i<n;i++) { for(int j =0;j<m;j++) { data[i][j]= in.nextInt(); } } int t = 5; long ans = dfs(0,0,-1,0); System.out.println(ans); } private static long dfs(int x, int y, int max, int cnt) { int ans = 0; if(x==n||y==m||cnt>k) { return 0; } int cur = data[x][y]; if(x==n-1&&y==m-1&&cnt==k) { return 1; } if(x==n-1&&y==m-1&&cnt==k-1&&cur>max) { return 1; } if(cur>max) { ans += dfs(x+1,y,cur,cnt+1); ans += dfs(x,y+1,cur,cnt+1); } ans += dfs(x+1,y,max,cnt); ans += dfs(x,y+1,max,cnt); ans %= 1000000007; return ans; } }

一、下面是加记忆功能的dfs

import java.util.Scanner; public class 记忆型dfs_地宫取宝 { static int[][] data; private static int n; private static int m; private static int k; public static void main(String[] args) { for(int i=0;i<51;i++) { for(int j=0;j<51;j++) { for(int l=0;l<13;l++) { for(int o=0;o<13;o++) { c[i][j][l][o] = -1; } } } } Scanner in = new Scanner(System.in); n = in.nextInt(); m = in.nextInt(); k = in.nextInt(); data = new int[n][m]; for(int i=0;i<n;i++) { for(int j =0;j<m;j++) { data[i][j]= in.nextInt(); } } int t = 5; long ans = dfs(0,0,-1,0); System.out.println(ans); } static long c[][][][] = new long[51][51][14][13]; private static long dfs(int x, int y, int max, int cnt) { int ans = 0; if(x==n||y==m||cnt>k) { return 0; } System.out.println(x); System.out.println(y); System.out.println(max); System.out.println(cnt); if(c[x][y][max+1][cnt]!=-1) return c[x][y][max+1][cnt]; int cur = data[x][y]; if(x==n-1&&y==m-1&&cnt==k) { return 1; } if(x==n-1&&y==m-1&&cnt==k-1&&cur>max) { return 1; } if(cur>max) { ans += dfs(x+1,y,cur,cnt+1); ans += dfs(x,y+1,cur,cnt+1); } ans += dfs(x+1,y,max,cnt); ans += dfs(x,y+1,max,cnt); ans %= 1000000007; c[x][y][max+1][cnt] = ans; return ans; } }