问题描述
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>
int a[52][52];
int row, col, k;
long long count;
void dfs(int i,int j ,int max, int num)
{
if (num > k || i > row || j > col)
return;
if (num == k && i==row && j==col)
{
count++;
return;
}
if (num == k - 1 && i == row && j == col && a[row][col] > max)
{
count++;
return;
}
for (int x = 0; x < 2; x++) //第一层循环,下移还是右移,0下移,1右移
{
for (int y = 0; y < 2; y++) //第二层循环,捡还是不捡,0不捡,1捡
{
if (a[i][j] <= max)
{
if (x)
{
dfs(i, j + 1, max, num);
break;
}
else
{
dfs(i + 1, j, max, num);
break;
}
}
else if (!y)
{
if (x)
dfs(i, j + 1, max, num);
else
dfs(i + 1, j, max, num);
}
else
{
if (x)
dfs(i, j + 1, a[i][j], num+1);
else
dfs(i + 1, j, a[i][j], num+1);
}
}
}
}
int main()
{
std::ios::sync_with_stdio(false);回
std::cin.tie(0);
std::cin >> row >> col >> k;
for (int i = 1; i <= row; i++)
for (int j = 1; j <= col; j++)
std::cin >> a[i][j];
dfs(1, 1, -1, 0);
std::cout << count % 1000000007;
return 0;
}
这回学乖了,很快就想到是用DFS,代码写得也较之前快,差不多花了20分钟这样子吧。(PS:递归写得真舒服,不知不觉写着写着就发现自己写完了),但是debug前前后后花了4个多小时…附上错误的原因吧
/*
//bug
for (int y = 0; y < 2; y++) //第二层循环,捡还是不捡,0不捡,1捡
{
if (!y)
{
if (!x)
dfs(i + 1, j, max ,num);
else
dfs(i, j + 1, max ,num);
}
else
{
if (a[i][j] > max)
{
if (!x)
dfs(i + 1, j, a[i][j],num+1);
else
dfs(i, j + 1, a[i][j],num+1);
}
else
{
if (!x)
dfs(i + 1, j, max,num);
else
dfs(i, j + 1, max,num);
}
}
*/
为啥自己出错了呢…因为还是这个二重循环有关,分为4种情况,捡,不捡与右走下走相互独立,看着挺好的,但是代码实现就有问题了,决定捡情况中如果金额小不能捡呢?这不就相当于不捡了么?这不就重复了么?
意识到这个问题后自己又改了下,还是错了,还是犯了上面那个错误,最后改成现在这样了:
当你没资格捡:你就只能下或右走;
你有资格捡:你可以选择捡或不捡以及下或右走;
这就形成最开始的那个代码,看上去有点没逻辑…实际上能运行还挺好的
只不过最后4个例子超时了,好像要用DP什么的,但是目前还没学,以后有机会有能力改