动态规划:1269. 停在原地的方案数(困难)

有一个长度为 arrLen 的数组,开始有一个指针在索引 0 处。

每一步操作中,你可以将指针向左或向右移动 1 步,或者停在原地(指针不能被移动到数组范围外)。

给你两个整数 steps 和 arrLen ,请你计算并返回:在恰好执行 steps 次操作以后,指针仍然指向索引 0 处的方案数。

由于答案可能会很大,请返回方案数 模 10^9 + 7 后的结果。

示例 1:输入:steps = 3, arrLen = 2

              输出:4

              解释:3 步后,总共有 4 种不同的方法可以停在索引 0 处。

                        向右,向左,不动

                        不动,向右,向左

                        向右,不动,向左

                        不动,不动,不动

示例  2:输入:steps = 2, arrLen = 4

               输出:2

               解释:2 步后,总共有 2 种不同的方法可以停在索引 0 处。

                          向右,向左

                          不动,不动


解题思路:使用动态规划:

                1.确定状态:dp[i][j]为在i步操作下到达下标j的方案数;

                2.确定状态转移方程:dp[i][j] = dp[i-1][j-1] + dp[i-1][j] + dp[i-1][j+1]

                3.确定开始条件及边界条件:dp[0][0] = 1; dp[0][j>0] = 0; 0<=i<=steps; 0<=j<=steps or arrlen.

                4.计算顺序:dp[0][0]——>dp[1][0]——>dp[1][1]——>dp[1][2]......

class Solution {

    public int numWays(int steps, int arrLen) {

        final int MODULO = 1000000007;

        int maxIndex = Math.min(steps, arrLen-1);

        int[][] dp = new int[steps+1][maxIndex+1];

        dp[0][0] = 1;

        for(int i=1; i<=steps; i++) {

            for(int j=0; j<=maxIndex; j++) {

                dp[i][j] = dp[i-1][j];

                if(j+1 <= maxIndex) {

                    dp[i][j] =  (dp[i][j] + dp[i - 1][j + 1]) % MODULO;

                }

                if(j-1 >= 0) {

                    dp[i][j] = (dp[i][j] + dp[i - 1][j - 1]) % MODULO;

                }

            }

        }

        return dp[steps][0];

    }

}

                

        

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容