【历届试题 地宫取宝】(dfs+记忆化)

时间限制: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

分析:
看到题目,第一感觉比较像数字塔(只能向右或向下走),然后到一个位置后,若当前位置的物品价值大于之前所有的就可以选择拿与不拿, 寻找最终到达右下角时物品总数等于k的方案数,这又像是用dp
数据范围:n,m<=50,1<k<=12,0<=c<=12
如果只是用dfs的话,递归层数最多能达到50层,会T,然后数据范围不大,并且物品最大值为12,结合题意,
可以用dp[x][y][cnt][maxx]:(x,y) 当前位置已经有cnt个物品,最大的物品价值为maxx
详见代码:

#include <bits/stdc++.h>
using namespace std;

#define mem(a,n) memset(a,n,sizeof(a))
#define memc(a,b) memcpy(a,b,sizeof(b))
#define rep(i,a,n) for(int i=a;i<n;i++) ///[a,n)
#define dec(i,n,a) for(int i=n;i>=a;i--)///[n,a]
#define pb push_back
#define fi first
#define se second
#define IO ios::sync_with_stdio(false)
#define fre freopen("in.txt","r",stdin)
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef long long ll;
typedef unsigned long long ull;
const double PI=acos(-1.0);
const double E=2.718281828459045;
const double eps=1e-3;
const int INF=0x3f3f3f3f;
const int MOD=1e9+7;
const int N=50+5;
const ll maxn=1e6+5;
const int dir[4][2]= {-1,0,1,0,0,-1,0,1};
int n,m,k;
int dp[N][N][13][13];
int a[N][N];
int dfs(int x,int y,int cnt,int maxx)
{
    int& ret=dp[x][y][cnt][maxx+1];
    if(ret!=-1)///已经访问过
        return ret;
    int num=0;
    if(x==n&&y==m)///递归终止条件
    {
        if(cnt==k||a[x][y]>maxx&&cnt==k-1)
            num++;
        return ret=num;///记忆并返回
    }
    if(x+1<=n)///右行
    {
        if(a[x][y]>maxx)///选择拿起当前位置的物品
        {
            num=(num+dfs(x+1,y,cnt+1,a[x][y]))%MOD;
        }
        ///选择不拿
        num=(num+dfs(x+1,y,cnt,maxx))%MOD;
    }
    if(y+1<=m)
    {
        if(a[x][y]>maxx)///选择拿
        {
            num=(num+dfs(x,y+1,cnt+1,a[x][y]))%MOD;
        }
        ///选择不拿
        num=(num+dfs(x,y+1,cnt,maxx))%MOD;
    }
    return ret=num;///记忆并返回
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        rep(i,1,n+1)
        {
            rep(j,1,m+1) scanf("%d",&a[i][j]);
        }
        mem(dp,-1);///初始化
        printf("%d\n",dfs(1,1,0,-1));
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值