蓝桥杯 PREV-28 地宫取宝

历届试题 地宫取宝  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　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

分析：一开始直接深搜的话会超时，所以采用记忆化搜索，减少重复的计算，节省时间。

代码：

#include<iostream>
#include<algorithm>
#include<vector>
#define M 1000000007
using namespace std;
long long dp[51][51][14][13];
vector<vector<int> > v;
int n, m, k;
long long dfs(int x, int y, int value, int cnt) {
	if (dp[x][y][value + 1][cnt] != -1) {
		return dp[x][y][value + 1][cnt];
	}
	if (x >= n || y >= m || cnt > k) {
		return 0;
	}
	long long ans = 0;
	if (x == n - 1 && y == m - 1) {
		if (cnt == k || cnt == k - 1 && v[x][y] > value) {
			ans++;
			ans %= M;
		}
		dp[x][y][value + 1][cnt] = ans;
		return ans;
	}
	if (x + 1 < n) {
		if (v[x][y] > value) {
			ans += dfs(x + 1, y, v[x][y], cnt + 1);
			ans %= M;
		}
		ans += dfs(x + 1, y, value, cnt);
		ans %= M;
	}
	if (y + 1 < m) {
		if (v[x][y] > value) {
			ans += dfs(x, y + 1, v[x][y], cnt + 1);
			ans %= M;
		}
		ans += dfs(x, y + 1, value, cnt);
		ans %= M;
	}
	dp[x][y][value + 1][cnt] = ans;
	return ans;
}
int main() {
	fill(dp[0][0][0], dp[0][0][0] + 51 * 51 * 14 * 13, -1);
	cin >> n >> m >> k;
	v.resize(n + 1);
	for (int i = 0; i <= n; i++) {
		v[i].resize(m + 1);
	}
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < m; j++) {
			cin >> v[i][j];
		}
	}
	cout << dfs(0, 0, -1, 0) << endl;
	return 0;
}