蓝桥杯 地宫寻宝 DFS 动态规划

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <queue>
using namespace std;

const int maxn = ;
const long long INF = ; //用来取余
int maze[maxn][maxn];

int d[][][][]; //行,列,k个数,value
int n, m, k;           //[n,m], k件宝贝
long long ans;         //方案数 

int dfs(int r, int c, int sum, int Max); //当前位置
void input();
void solve();

void input()
{
    memset(d, -, sizeof(d));
    scanf("%d%d%d", &n, &m, &k);
    for (int i = ; i < n; i++)
    {
        for (int j = ; j < m; j++) {
            scanf("%d", &maze[i][j]);
        }
    }
}

int dfs(int r, int c, int sum, int Max)
{
    if (d[r][c][sum][Max + ] != -) {      //已经遍历完, 并设置了结果
        return d[r][c][sum][Max + ];       //返回结果
    }

    int t = ;

    if (r == n -  && c == m - ) {         //到达入口 

        if (maze[r][c] > Max) {             //可以再拿一个宝物
            if (sum == k || sum == k - )   //如果已经到了 k 或是  k-1，方案++
                t++;
        }
        else if (sum == k) {                //不能拿时候,则此时就需要为k, 方案++
            t++;
        }
        return d[r][c][sum][Max + ] = t;   //更新方案数
    }

    if (r +  < n) {
        if (maze[r][c] > Max) {            //可以拿
            t += dfs(r + , c, sum + , maze[r][c]);      //选择拿
            t %= INF;
            t += dfs(r + , c, sum, Max);                 //选择不拿
            t %= INF;
        }
        else {
            t += dfs(r + , c, sum, Max);  //不可以拿
            t %= INF;
        }
    }
    if (c +  < m) {
        if (maze[r][c] > Max) {
            t += dfs(r, c + , sum + , maze[r][c]);
            t %= INF;
            t += dfs(r, c + , sum, Max);
            t %= INF;
        }
        else {
            t += dfs(r, c + , sum, Max);
            t %= INF;
        }
    }
    d[r][c][sum][Max + ] = t;         //将方案数 保存在Max + 1 处
    return d[r][c][sum][Max + ];      //返回方案数
}

void solve()
{
    input();
    ans = dfs(, , , -);
    cout << ans << endl;
}

int main()
{
    solve();
    return ;
}

//不太熟悉动态规划的题目.................,参考了网上解法,回头看看这类题目.......