【题目】
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
【资源约定】
峰值内存消耗 < 256M
CPU消耗 < 1000ms
【注意点】
1、看清题目看清题目看清题目!n,m都小于50(没有无穷大的!)
2、理解题目理解题目理解题目!是行动方案,同一条路径上取不同的为两种方案,记得看案例理解一下
3、这里不会回头,所以不需要vis数组
【方法一】
深度搜索,其实就是套公式,但是这个方法会超时
#include<iostream>
using namespace std;
int m,n,k;
int val[50][50];
bool legal(int a,int b)
{
if(a<0)
return false;
if(b<0)
return false;
if(a>=n)
return false;
if(b>=m)
return false;
return true;
}
long int ans=0;
const int x[2]={0,1};
const int y[2]={1,0};
void dfs(int a,int b,int have,int max)
{
int i,p,q;
if(a==n-1&&b==m-1)
if(have==k||(have==k-1&&max<val[a][b]))
{
ans++;
ans=ans>1000000007?ans%1000000007:ans;
return;
}
for(i=0;i<2;i++)
{
p=a+x[i];
q=b+y[i];
if(legal(p,q))
{
if(val[a][b]>max)
dfs(p,q,have+1,val[a][b]);
dfs(p,q,have,max);
}
}
}
int main()
{
cin>>n>>m>>k;
int i,j;
for(i=0;i<n;i++)
for(j=0;j<m;j++)
cin>>val[i][j];
dfs(0,0,0,-1);
cout<<ans;
return 0;
}
如果超时了,就要考虑剪枝,可以用个数组记录,避免重复计算
【方法二】
记忆搜索,记忆搜索是从后往前搜索,a可以到b,c,先算b,c达成目的的方式,记录下来,这两者相加就是a达成目的的方式,和深度搜索的想法恰好相反
#include<iostream>
#include<Cstring>
using namespace std;
int m,n,k;
int val[50][50];
long int ans[50][50][13][14];//a,b,have,max+1
bool legal(int a,int b)
{
if(a<0)
return false;
if(b<0)
return false;
if(a>=n)
return false;
if(b>=m)
return false;
return true;
}
//long int ans=0;
const int x[2]={0,1};
const int y[2]={1,0};
long int dfs(int a,int b,int have,int max)
{
if(ans[a][b][have][max+1]!=-1) return ans[a][b][have][max+1];
int i,p,q;
if(a==n-1&&b==m-1)
{
if(have==k||(have==k-1&&max<val[a][b]))
{
ans[a][b][have][max+1]=1;
return 1;
}
ans[a][b][have][max+1]=0;
return 0;
}
if(have>k) return 0;
long int cc=0;
for(i=0;i<2;i++)
{
p=a+x[i];
q=b+y[i];
if(legal(p,q))
{
if(val[a][b]>max)
cc=(cc+dfs(p,q,have+1,val[a][b]))%1000000007;
cc=(cc+dfs(p,q,have,max))%1000000007;
}
}
ans[a][b][have][max+1]=cc;
return cc;
}
int main()
{
memset(ans,-1,sizeof(ans));
cin>>n>>m>>k;
int i,j;
for(i=0;i<n;i++)
for(j=0;j<m;j++)
cin>>val[i][j];
//dfs(0,0,0,-1);
cout<<dfs(0,0,0,-1);
return 0;
}
这个方法刚开始我遗漏了
ans[a][b][have][max+1]=0;
return 0;
并且ans数组初始化时0,实际上ans里面的0也是有效的,这就导致了也可以得出实例里面的正确答案,但是会超时,所以这点一定要注意!