力扣 1269. 停在原地的方案数 dp

https://leetcode-cn.com/problems/number-of-ways-to-stay-in-the-same-place-after-some-steps/
在这里插入图片描述
思路: d p i , j dp_{i,j} dpi,j表示经过 i i i步后到达 j j j位置的方法数,显然 d p 0 , 0 = 1 dp_{0,0}=1 dp0,0=1,且有以下转移方程:
d p i , j = d p i − 1 , j − 1 + d p i − 1 , j + d p i − 1 , j + 1 dp_{i,j}=dp_{i-1,j-1}+dp_{i-1,j}+dp_{i-1,j+1} dpi,j=dpi1,j1+dpi1,j+dpi1,j+1
注意取模和位置的有效性即可。

class Solution {
public:
    int numWays(int steps, int arrLen) {
        vector<vector<int>> dp(steps+1,vector<int>(steps+1));
        dp[0][0]=1;
        const int mod=1e9+7;
        for(int i=1;i<=steps;i++)
        {
            int maxj=min(i,arrLen-1);
            for(int j=0;j<=maxj;j++)
            {
                if(j-1>=0)
                    dp[i][j]=(dp[i-1][j-1]+dp[i][j])%mod;
                if(j<maxj)
                    dp[i][j]=(dp[i-1][j+1]+dp[i][j])%mod;
                dp[i][j]=(dp[i-1][j]+dp[i][j])%mod;
            }
        }
        return dp[steps][0];
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值