2014年第五届蓝桥杯C++B组I题

2014年第五届蓝桥杯C++B组I题

地宫取宝

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

这道题记忆化搜索。
原本想想简单的dfs的。。后来是50*50的数据让我望而生畏。。还是老实记忆化搜索把。。

只能向下或者向右。
然后记录最大值就行。如果当前格子的值比拿在手上的大。（严格大）
那么有两种选择。拿或者不拿。
如果小的话，没得选择。只有不拿。。
所以
dp[x][y][sum][maxx]
记录表示当前点(x, y)拿了sum件物品。其中最大值为maxx

我在写代码的时候因为初始化dp为-1。然后那些格子中物品的价值有的可能为0，并且是严格大。所以开始dfs的时候maxx = -1；
所以代码部分dp的最后一维其实要比当前的最大价值大1
没有什么影响。。

代码中间把向下或者向右。取或者不取的方案数加起来就可以啦~

嘿嘿。还需多加练习呀

代码部分

#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a))
const int mod = 1e9 + 7;
const int N = 55;
using namespace std;

int n, m, k;
int a[N][N];
int dp[N][N][15][15];

int dfs(int x, int y, int sum, int maxx)
{
	if (dp[x][y][sum][maxx + 1] != -1)
	{
		return dp[x][y][sum][maxx + 1];
	}
	int ans = 0;
	if (x == n - 1 && y == m - 1)
	{
		if (sum == k || sum == k - 1 && a[x][y] > maxx)
		{
			ans++;
			ans %= mod;
		}
		return dp[x][y][sum][maxx + 1] = ans % mod;
	}
	if (x < n - 1)
	{
		if (a[x][y] > maxx)
		{
			ans += dfs(x + 1, y, sum + 1, a[x][y]);
			ans %= mod;
		}
		ans += dfs(x + 1, y, sum, maxx);
		ans %= mod;
	}
	if (y < m - 1)
	{
		if (a[x][y] > maxx)
		{
			ans += dfs(x, y + 1, sum + 1, a[x][y]);
			ans %= mod;
		} 
		ans += dfs(x, y + 1, sum, maxx);
		ans %= mod;
	}
	return dp[x][y][sum][maxx + 1] = ans % mod;
}

int main()
{
	cin >> n >> m >> k;
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < m; j++)
		{
			scanf ("%d", &a[i][j]);
		}
	}
	mst(dp, -1);
	cout << dfs(0, 0, 0, -1) << endl;
	return 0;
}