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

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

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

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

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

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

思路

用的是动态规划的方法，dp[i][j]，走i步后处在j点时的方案数

递推公式是：dp[i][j]=dp[i-1][j-1]+dp[i-1][j]+dp[i-1][j+1]

因为一次只能走1步（或不走）走i步到j点i-1步时的位置只有三种可能：j-1 , j , j+1，所以得到上述递推公式

特别的，当dp[i][j]所在的点是0,走i步到j点i-1步时的位置只有两种可能： j , j+1

同理当dp[i][j]所在的点是最远点,走i步到j点i-1步时的位置只有两种可能： j-1 , j

 

然后是数组长度的问题，虽然条件给的arrLen最大有10^6之多，但在每步只允许走一格的情况下，长度是不会超过步数的，因此可以将长度定义为：min(steps+1,arrLen)

 

代码

int min(int a,int b)
{
    return a<b?a:b;
}

int numWays(int steps, int arrLen)
{
         arrLen=min(steps+1,arrLen);
        int dp[steps+1][arrLen];
        int i,j;
        dp[0][0]=1;
       for(i=1;i<arrLen;i++)
       dp[0][i]=0;
       for(i=1;i<=steps;i++)
       for(j=0;j<arrLen;j++)
       {
           dp[i][j]=dp[i-1][j];
            if(j>=1)
            dp[i][j]=(dp[i-1][j-1]+dp[i][j])%1000000007;
            if(j+1<arrLen)
            dp[i][j]=(dp[i-1][j+1]+dp[i][j])%1000000007;
       }
       return dp[steps][0];
}