记忆化深搜——地宫取宝问题

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;
    }
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值