标题:地宫取宝
X 国王有一个地宫宝库。是 n x m 个格子的矩阵。每个格子放一件宝贝。每个宝贝贴着价值标签。
地宫的入口在左上角,出口在右下角。
小明被带到地宫的入口,国王要求他只能向右或向下行走。
走过某个格子时,如果那个格子中的宝贝价值比小明手中任意宝贝价值都大,小明就可以拿起它(当然,也可以不拿)。
当小明走到出口时,如果他手中的宝贝恰好是k件,则这些宝贝就可以送给小明。
请你帮小明算一算,在给定的局面下,他有多少种不同的行动方案能获得这k件宝贝。
数据格式
输入一行3个整数,用空格分开:n m k (1<=n,m<=50, 1<=k<=12)
接下来有 n 行数据,每行有 m 个整数 Ci (0<=Ci<=12)代表这个格子上的宝物的价值 价值可以取0
要求输出一个整数,表示正好取k个宝贝的行动方案数。该数字可能很大,输出它对 1000000007 取模的结果。
例如,输入:
2 2 2
1 2
2 1
程序应该输出:
2
再例如,输入:
2 3 2
1 2 3
2 1 5
程序应该输出:
14
cpp实现代码如下:(附优化前与优化后)
#include <iostream>
#include <cstring>
using namespace std;
int n, m, k;
int data[50][50];
long long ans;
const int mod = 1000000007;
long long cache[50][50][15][15];
void dfs(int x, int y, int max, int cnt) // 普通的dfs 会超时
{
if (x == n || y == m)
return;
int cur = data[x][y];
if (x == n - 1 && y == m - 1)
{
if (cnt == k || cnt == k - 1 && cur > max)
{
ans++;
if (ans > mod)
ans %= mod;
}
}
if (cur > max)
{
//遇到价值比当前手里最大值大的可以取
dfs(x, y + 1, cur, cnt + 1);
dfs(x + 1, y, cur, cnt + 1);
}
//当然也可以不取 或者遇到的价值没有当前手里的大
dfs(x, y + 1, max, cnt);
dfs(x + 1, y, max, cnt);
}
//优化后解法 dfs2
long long dfs2(int x, int y, int max, int cnt) // 改成记忆型递归,可以减少重复子问题的计算
{
if (cache[x][y][max + 1][cnt] != -1) //一开始传入max = -1 在这里可以直接加1 代表-1
return cache[x][y][max + 1][cnt]; //不等于-1说明历史上求解过,直接返回即可
long long ans = 0; //每次递归ans从0开始 注意这里不能使用全局变量ans
if (x == n || y == m || cnt > k)
return 0;
int cur = data[x][y];
if (x == n - 1 && y == m - 1)
{
if (cnt == k || cnt == k - 1 && cur > max)
{
ans++;
if (ans > mod)
ans %= mod;
}
return ans;
}
if (cur > max)
{
//遇到价值比当前手里最大值大的可以取
ans += dfs2(x, y + 1, cur, cnt + 1);
ans += dfs2(x + 1, y, cur, cnt + 1);
}
//当然也可以不取 或者遇到的价值没有当前手里的大
ans += dfs2(x, y + 1, max, cnt);
ans += dfs2(x + 1, y, max, cnt);
cache[x][y][max + 1][cnt] = ans % mod;
return cache[x][y][max + 1][cnt];
}
int main(int argc, char const *argv[])
{
memset(cache, -1, sizeof(cache)); //初始化cache标记数组
scanf("%d %d %d", &n, &m, &k);
for (int i = 0; i < n; i++)
for (int j = 0; j < m; j++)
scanf("%d", &data[i][j]);
// dfs(0, 0, -1, 0); //第一个点的价值可能是0,所以要从-1开始
// printf("%d\n", ans);
printf("%lli", dfs2(0, 0, -1, 0));
return 0;
}