LeetCode 576. 出界的路径数

最新推荐文章于 2021-08-23 15:18:14 发布

原创最新推荐文章于 2021-08-23 15:18:14 发布 · 1.1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #动态规划

LeetCode 专栏收录该内容

54 篇文章

订阅专栏

给定一个m×n网格和一个球，从(i,j)开始，最多移动N次，求出球能够超出边界的路径数。动态规划解法中，每个状态表示在指定位置且剩余步数为N的路径数，通过转移方程更新状态。当球出界后不能返回。示例和解题思路详细阐述了问题的求解过程。" 52088543,5083427,无向图连通分量遍历策略解析,"['图论', '图算法', '深度优先搜索', '广度优先搜索', '数据结构']

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

给定一个 m × n 的网格和一个球。球的起始坐标为 (i,j) ，你可以将球移到相邻的单元格内，或者往上、下、左、右四个方向上移动使球穿过网格边界。但是，你最多可以移动 N 次。找出可以将球移出边界的路径数量。答案可能非常大，返回结果 mod 109 + 7 的值。

示例 1：

输入: m = 2, n = 2, N = 2, i = 0, j = 0
输出: 6
解释:

示例 2：

输入: m = 1, n = 3, N = 3, i = 0, j = 1
输出: 12
解释:

说明:

球一旦出界，就不能再被移动回网格内。
网格的长度和高度在 [1,50] 的范围内。
N 在 [0,50] 的范围内。

思路：这道题需要使用动态规划来做，但是要知道每个状态是怎么转移的，对于一个起始点为i，j，N步可以走出的点的路径个数，等于该点周围的4个点，N-1步可以走出的路径个数之和，知道了这个之后，我们就可以以这个公式作为状态转移方程。

代码：

class Solution {
public:
    int findPaths(int m, int n, int N, int i, int j) {
        vector<vector<vector<int> > > dp(N+1, vector<vector<int> >(m, vector<int>(n)));
        long long MOD=1000000007;
        
        for(int k=1; k<=N; ++k){
            for(int x=0; x<m; ++x){
                 for(int y=0; y<n; ++y){
                    long long v1 = (x == 0) ? 1 : dp[k-1][x-1][y];
                    long long v2 = (y == 0) ? 1 : dp[k-1][x][y-1];
                    long long v3 = (x == m-1) ? 1 : dp[k-1][x+1][y];
                    long long v4 = (y == n-1) ? 1 : dp[k-1][x][y+1];
                    
                    dp[k][x][y]=(v1+v2+v3+v4)%MOD;
                }
            }
        }
        
        return dp[N][i][j];
    }
};