1269. 停在原地的方案数

最新推荐文章于 2023-08-15 23:34:13 发布

不叫芫荽

最新推荐文章于 2023-08-15 23:34:13 发布

阅读量109

点赞数

分类专栏： LeetCode刷题

本文链接：https://blog.csdn.net/CRJ297486/article/details/116763475

版权

LeetCode刷题专栏收录该内容

94 篇文章 1 订阅

订阅专栏

该博客探讨了一个编程问题，涉及在一个给定长度的数组中，计算在执行一定次数的操作后，指针仍回到初始位置的方法数。通过动态规划方法，博主详细解释了如何计算在不超过数组长度和步数限制的情况下，所有可能的路径，并返回结果模10^9+7后的值。示例展示了不同步骤和数组长度下的解决方案计数。

摘要由CSDN通过智能技术生成

有一个长度为 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 处。
向右，向左
不动，不动

示例 3：

输入：steps = 4, arrLen = 2
输出：8

提示：

1 <= steps <= 500
1 <= arrLen <= 10^6

class Solution {
public:
    int numWays(int steps, int arrLen) {
        int mod=1000000007;
        int len=min(arrLen,steps+1);
        //vector<vector<int>>dp(steps+1,vector<int>(len));
//        int dp[steps+1][arrLen+2];
        vector<int>dp(len);
        vector<int>dp2(len);
        dp[0]=1;
        for(int i=1;i<=steps;i++){
            for(int j=0;j<len;j++){
                dp2[j]=dp[j];
                if(j>0)dp2[j]=(dp2[j]+dp[j-1])%mod;
                if(j<len-1)dp2[j]=(dp2[j]+dp[j+1])%mod;
            }
            dp=dp2;
            //cout<<endl;
        }
        return dp[0];

    }
};