蓝桥杯 - 地宫取宝

用暴力搜索会超时,改为记忆化搜索,就是DP。

#include <cstdio>
#include <string.h>

#define MAX 50 + 10
#define MOD 1000000007

int n, m, k;
int MGraph[MAX][MAX];
int dp[MAX][MAX][MAX][MAX];
//int ans = 0;

/*
void dfs( int curX, int curY, int curSum, int curMax ) {
    if( curX >= n || curY >= m ) return;    // 剪枝
    if( curSum > k ) return;    // 剪枝

    // 走到终点并找到解
    if( curX == n - 1 && curY == m - 1 ) {
        if( ( curSum == k ) || ( ( curSum == k - 1 ) && MGraph[curX][curY] > curMax ) )
            ans = ( ans + 1 ) % 1000000007;
        return;
    }
    // 如果可以拿
    if( MGraph[curX][curY] > curMax ) {
        // 拿
        dfs( curX + 1, curY, curSum + 1, MGraph[curX][curY] );
        dfs( curX, curY + 1, curSum + 1, MGraph[curX][curY] );
        // 不拿
        dfs( curX + 1, curY, curSum, curMax );  // 向下
        dfs( curX, curY + 1, curSum, curMax );  // 向右
    }
    // 不能拿
    else {
        dfs( curX + 1, curY, curSum, curMax );  // 向下
        dfs( curX, curY + 1, curSum, curMax );  // 向右
    }

    if( curX + 1 < n ) {
        if( MGraph[curX][curY] > curMax ) {
            dfs( curX + 1, curY, curSum + 1, MGraph[curX][curY] );
        }
        dfs( curX + 1, curY, curSum, curMax );
    }
    if( curY + 1 < m ) {
        if( MGraph[curX][curY] > curMax ) {
            dfs( curX, curY + 1, curSum + 1, MGraph[curX][curY] );
        }
        dfs( curX, curY + 1, curSum, curMax );
    }

}
*/

int dfs( int curX, int curY, int curSum, int curMax ) {
    if( dp[curX][curY][curSum][curMax] != -1 ) {
        return dp[curX][curY][curSum][curMax];
    }

    if( curX >= n || curY >= m ) return 0;    // 剪枝
    if( curSum > k ) return 0;    // 剪枝

    // 走到终点并找到解
    if( curX == n - 1 && curY == m - 1 ) {
        if( ( curSum == k ) || ( ( curSum == k - 1 ) && MGraph[curX][curY] > curMax ) )
            return dp[curX][curY][curSum][curMax] = 1;
        return dp[curX][curY][curSum][curMax] = 0;
    }

    int t = 0;
    // 如果可以拿
    if( MGraph[curX][curY] > curMax ) {
        // 拿
        t = ( t + dfs( curX + 1, curY, curSum + 1, MGraph[curX][curY] ) ) % MOD;
        t = ( t + dfs( curX, curY + 1, curSum + 1, MGraph[curX][curY] ) ) % MOD;
        // 不拿
        t = ( t + dfs( curX + 1, curY, curSum, curMax ) ) % MOD;  // 向下
        t = ( t + dfs( curX, curY + 1, curSum, curMax ) ) % MOD;  // 向右
    }
    // 不能拿
    else {
        t = ( t + dfs( curX + 1, curY, curSum, curMax ) ) % MOD;  // 向下
        t = ( t + dfs( curX, curY + 1, curSum, curMax ) ) % MOD;  // 向右
    }

    return dp[curX][curY][curSum][curMax] = t % MOD;
}

int main() {
    memset( dp, -1 , sizeof( dp ) );
    scanf( "%d%d%d", &n, &m, &k );

    for( int i = 0; i < n; i++ ) {
        for( int j = 0; j < m; j++ ) {
            scanf( "%d", &MGraph[i][j] );
            MGraph[i][j]++;
        }
    }

    int ans = dfs( 0, 0, 0, 0 );
    printf( "%d\n", ans );
    return 0;
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值