蓝桥杯第五届A组真题-地宫取宝（动态规划）

蓝桥杯第五届A组真题-地宫取宝（动态规划）+Java代码实现

题目描述

---

dp思路

这里的dp重点去考虑几维问题，首先物品是n*m的方格，因此要走完所有物品即需要二维遍历。

其次要判断当前拿的物品个数，又要加一维。

最后重点去考虑拿不拿当前物品，而要拿起当前物品的前提条件是：当前物品的价值比之前所拿的任何物品都要大，因此我们需要用一维去存储所拿的最后一个物品的价值。即当前手中物品的最大价值。

以上分析可得我们共需要四维数组遍历，且这个f[i][j][k][c]四维数组所表示的即从开头第1行1列走到当前第i行第j列时手里拿了k个物品且最后一个物品的最大价值为c的集合，它的值为这个集合的总个数

另外要注意
我们在遍历第四位，即当前走到的这个物品的价值时，要拿起这个物品时需判断手里的最大价值是否小于当前物品的价值，而这是需要用到循环，只要手里的最大价值不超过这个即可拿起当前物品，因此我们在拿起物品时仍然需要一次循环。

具体分析图如下：

Java代码

import java.util.Scanner;

public class Main {
    static int n;
    static int m;
    static int cnt;
    static int[][][][] f;//状态表示：f[i,j,k,c]表示从初始位置走到第i行第j列手拿k件物品且最后一件物品的价值是c的集合，属性为该状态总个数。
    static int[][] w;//物品价值
    final static int MOD = 1000000007;

    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        n = scan.nextInt();
        m = scan.nextInt();
        cnt = scan.nextInt();
        w = new int[n + 5][m + 5];
        f = new int[n + 5][m + 5][cnt + 5][15];
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                w[i][j] = scan.nextInt();
                w[i][j]++;//物品的价值为1~13，以便用0表示当前没有任何所拿的物品
            }
        }

        //初始化第一个位置的两种情况，分别为拿第一个与不拿第一个均有一种行动方案
        f[1][1][1][w[1][1]] = 1;
        f[1][1][0][0] = 1;

        //寻找走到第i行第j列时手拿k件物品且最后一个物品价值为c的情况集合总个数
        for (int i = 1; i <= n; i++) {
            for (int j = 1; j <= m; j++) {
                for (int k = 0; k <= cnt; k++) {
                    for (int c = 0; c < 15; c++) {
                        //不拿当前第[i,j]件物品
                        f[i][j][k][c] = (f[i][j][k][c] + f[i - 1][j][k][c]) % MOD;//取模操作千万别忘记
                        f[i][j][k][c] = (f[i][j][k][c] + f[i][j - 1][k][c]) % MOD;

                        //拿当前第[i,j]件物品，要保证当前物品的价值为c，且为了防止数组溢出要保证k>0
                        if (w[i][j] == c && k > 0) {
                            for (int l = 0; l < c; l++) {
                                f[i][j][k][c] = (f[i][j][k][c] + f[i - 1][j][k - 1][l]) % MOD;
                                f[i][j][k][c] = (f[i][j][k][c] + f[i][j - 1][k - 1][l]) % MOD;
                            }
                        }
                    }
                }
            }
        }
        int num = 0;
        for (int i = 1; i < 15; i++) {
            num = (num + f[n][m][cnt][i]) % MOD;
        }
        System.out.println(num);

    }
}