[蓝桥杯 2014 省 AB] 地宫取宝
题目描述
X 国王有一个地宫宝库。是 n × m n \times m n×m 个格子的矩阵。每个格子放一件宝贝。每个宝贝贴着价值标签。
地宫的入口在左上角,出口在右下角。
小明被带到地宫的入口,国王要求他只能向右或向下行走。
走过某个格子时,如果那个格子中的宝贝价值比小明手中任意宝贝价值都大,小明就可以拿起它(当然,也可以不拿)。
当小明走到出口时,如果他手中的宝贝恰好是 k k k 件,则这些宝贝就可以送给小明。
请你帮小明算一算,在给定的局面下,他有多少种不同的行动方案能获得这 k k k 件宝贝。
输入格式
输入一行 3 3 3 个整数,用空格分开: n n n, m m m, k ( 1 ≤ n , m ≤ 50 , 1 ≤ k ≤ 12 ) k(1 \le n,m \le 50,1 \le k \le 12) k(1≤n,m≤50,1≤k≤12)。
接下来有 n n n 行数据,每行有 m m m 个整数 C i ( 0 ≤ C i ≤ 12 ) C_i(0 \le C_i \le 12) Ci(0≤Ci≤12) 代表这个格子上的宝物的价值。
输出格式
要求输出一个整数,表示正好取 k k k 个宝贝的行动方案数。该数字可能很大,输出它对 1000000007 ( 1 0 9 + 7 ) 1000000007(10^9+7) 1000000007(109+7) 取模的结果。
样例 #1
样例输入 #1
2 2 2
1 2
2 1
样例输出 #1
2
样例 #2
样例输入 #2
2 3 2
1 2 3
2 1 5
样例输出 #2
14
提示
时限 1 秒, 256M。蓝桥杯 2014 年第五届省赛
Java代码
import java.util.*;
class Main
{
public final static int MOD = 1000000007;
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[][] w = new int[n+1][m+1];
int[][][][] dp =new int[n+1][m+1][13][14]; //四个状态分别表示所在位置、当前拿的宝物数目、手中宝物最大价值
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
w[i][j]=sc.nextInt()+1; //价值从一开始,即把所有宝物价值加一,方便将0作为最小宝物价值
dp[1][1][1][w[1][1]]=1; //选第一个
dp[1][1][0][0]=1; //不选第一个
for(int i=1;i<=n;i++)
{
for(int j=1;j<=m;j++)
{
if(i==1 && j==1) continue; //最开始位置已经初始化了
for(int u=0;u<=k;u++)
{
for(int v=0;v<=13;v++)
{
dp[i][j][u][v] = (dp[i][j][u][v] + dp[i-1][j][u][v])%MOD; //从上往下走不取当前宝物
dp[i][j][u][v] = (dp[i][j][u][v] + dp[i][j-1][u][v])%MOD; //从左往右走不取当前宝物
if(u>=1 && v==w[i][j]) //如果手中宝物数目大于等于1且最大价值为当前格子宝物价值才可以拿
{
for(int c=0;c<v;c++) //遍历所有比当前宝物价值小的价值作为当前状态的前一个状态
{
dp[i][j][u][v] = (dp[i][j][u][v] + dp[i][j-1][u-1][c])%MOD; //从上往下走取当前宝物
dp[i][j][u][v] = (dp[i][j][u][v] + dp[i-1][j][u-1][c])%MOD; //从左往右走取当前宝物
}
}
}
}
}
}
int cnt=0;
for(int i=0;i<=13;i++) cnt=(cnt+dp[n][m][k][i])%MOD; //所有最大价值的取法
System.out.println(cnt%MOD);
}
}