记忆化深搜——地宫取宝问题
java实现:
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Scanner;
public class Main {
static int N=1000000007;
static long [][][][]visit=new long[50][50][15][15];//记忆化存储,分别有四个属性 x,y,num:方案数,max:当前的最大价值
public static void main(String[] args) {
Scanner sc=new Scanner(System.in);
int n=sc.nextInt();
int m=sc.nextInt();
int k=sc.nextInt();
int [][]map=new int[n][m];
initvisit();
for(int i=0;i<n;i++) {
for(int j=0;j<m;j++) {
map[i][j]=sc.nextInt();
}
}
dfs(map,k,0,0,0,-1);
System.out.println(visit[0][0][0][0]);
}
public static void initvisit() {
for(int a=0;a<50;a++) {
for(int b=0;b<50;b++) {
for(int c=0;c<15;c++) {
for(int d=0;d<15;d++) {
visit[a][b][c][d]=-1;
}
}
}
}
}
public static long dfs(int map[][],int k,int x,int y,int num,int max) {
if(x<map.length&&y<map[0].length&&num<=k) {
if(visit[x][y][num][max+1]!=-1) {
return visit[x][y][num][max+1];
}
else {
if(x==map.length-1&&y==map[0].length-1) {
if(k==num||(k==num+1&&map[x][y]>max)) {
visit[x][y][num][max+1]=1;
return visit[x][y][num][max+1];
}
}
else {
if(map[x][y]>max) { //当前宝物价值比手中任何一个都大
int t=map[x][y];
visit[x][y][num][max+1]=dfs(map,k,x+1,y,num,max)%N+dfs(map,k,x+1,y,num+1,t)%N
+dfs(map,k,x,y+1,num,max)%N+dfs(map,k,x,y+1,num+1,t)%N;
//分为四种情况继续搜索 : 1.拿起向右走 2.不拿向右走 3.拿起向下走 4.不拿向下走
}
else {
visit[x][y][num][max+1]=dfs(map,k,x+1,y,num,max)%N+dfs(map,k,x,y+1,num,max)%N;
//分为两种情况继续搜索 :1.不拿向右走 2.不拿向下走
}
return visit[x][y][num][max+1]%N;
}
}
}
return visit[x][y][num][max+1]=0;
}
}