蓝桥杯 地宫寻宝(DP or DP+记忆化搜索)

蓝桥杯  地宫寻宝

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,否则按无效代码处理。

题解:

  先%一发我伟大的队长 杰哥tql

  引入:m×n的方格,从左下角走到右上角,总共的路径为C(m+n) m = C(m+n) n

 1 #include<cstdio>
 2 #define ll long long
 3 int val[58][58];
 4 ll C[118][118],dp[58][58][18]={0},mod=1000000007;
 5 //dp[x][y][k]表示x,y位置上的宝物作为第k个被拿的方案数 
 6 void init()//求组合数 
 7 {
 8     C[0][0]=1;
 9     for(int i=1;i<=100;i++)
10     {
11         C[i][0]=1;
12         for(int j=1;j<=i;j++)
13         {
14             if(j<=i/2)
15                 C[i][j]=(C[i-1][j]+C[i-1][j-1])%mod;
16             else
17                 C[i][j]=C[i][i-j]; 
18         }
19     }
20 }
21 int main()
22 {
23     init();
24     int n,m,K;
25     scanf("%d%d%d",&n,&m,&K);
26     for(int i=0;i<n;i++)
27         for(int j=0;j<m;j++)
28             scanf("%d",&val[i][j]);
29     ll ans=0;
30     for(int i=0;i<n;i++)
31         for(int j=0;j<m;j++)
32         {
33             //i,j位置上的宝物是第一个被拿的方案数即从开始地点走到i,j位置的方案数 
34             dp[i][j][1]=C[i+j][i];
35             for(int k=i;k>=0;k--)//这两层循环表示i,j位置可以由哪些位置走到 
36                 for(int l=j;l>=0;l--)
37                 {
38                     if(val[k][l]<val[i][j])
39                         for(int t=2;t<=K;t++)
40                             dp[i][j][t]=(dp[i][j][t]+(dp[k][l][t-1]*C[i-k+j-l][i-k])%mod)%mod;//C[i-k+j-l][i-k]相当于从k,l位置走到i,j位置的方案数 
41                 }
42             ans=(ans+(dp[i][j][K]*C[n-1-i+m-1-j][n-1-i])%mod)%mod;//从i,j位置到n-1,m-1位置的方案数 
43         }    
44     printf("%lld\n",ans);
45     return 0;
46 }

以及大众化的记忆化搜索写法:

 1 import java.util.Scanner;
 2 
 3 public class Main {
 4     static int[][][][] dp = new int[55][55][15][15];
 5     static int[][] a = new int[55][55];
 6     static int[][] vis = new int[55][55];
 7     static int[][] dir =  {{0,1},{1,0}};
 8     static int mod = 1000000007;
 9     static int n,m,k; 
10     static int dfs(int x,int y,int step,int maxv) {
11         if(dp[x][y][step][maxv]!=-1) 
12             return dp[x][y][ step][maxv];
13         int cnt = 0;
14         if(x==n-1&&y==m-1) { 
15             if((step == k-1&&maxv<a[x][y])||(step==k)) {
16                 return dp[x][y][step][maxv] = 1;
17             }
18             return dp[x][y][step][maxv] = 0;
19         }
20         for(int i=0;i<2;i++) {
21             int tx = x+dir[i][0];
22             int ty = y+dir[i][1];
23             if(tx<0||ty<0||tx>=n||ty>=m||vis[tx][ty]==1) {
24                 continue;
25             }
26             vis[tx][ty] = 1;
27             if(a[x][y]>maxv) {
28                 cnt = (cnt+dfs(tx, ty, step+1, a[x][y]))%mod;
29             }
30             cnt = (cnt + dfs(tx, ty, step, maxv))%mod;
31             vis[tx][ty] = 0;
32         }
33         return dp[x][y][step][maxv] = cnt%mod;
34     }
35     public static void main(String[] args) {
36         Scanner cin = new Scanner(System.in);
37         n = cin.nextInt();
38         m = cin.nextInt();
39         k = cin.nextInt();
40         for(int i=0;i<55;i++)
41             for(int j=0;j<55;j++)
42                 for(int k=0;k<15;k++)
43                     for(int l=0;l<15;l++)
44                         dp[i][j][k][l] = -1;
45         for(int i=0;i<n;i++)
46             for(int j=0;j<m;j++) {
47                 a[i][j] = cin.nextInt();
48                 a[i][j]++;
49             }
50                 
51         
52         System.out.println(dfs(0,0,0,0));
53     }
54 
55 }

猜你喜欢

转载自www.cnblogs.com/1013star/p/10332071.html