题解 | #圆环回原点#

圆环回原点

http://www.nowcoder.com/questionTerminal/16409dd00ab24a408ddd0c46e49ddcf8

dp定义

dp[i][j]表示走i步到达编号为j的节点共有多少中方法

状态转移

dp[i][j] = dp[i-1][j-1] (i-1步走到j左边的方法数) + dp[i-1][j+1](i-1步走到j右边的方法数)

注意: 上边为了方便理解,没有处理j-1和j+1的越界问题,在下边代码中体现

base case

注意到状态转移需要上一行的数据,所以需要初始化0行。

dp[0][0] = 1, 其余dp[0][...]初始化为0.

代码

class Solution {
public:
    int circle(int n) { //求走n步回到原点的方法数
        int mod = 1000000007; //结果可能特别大,要对mod取余
        int K = 10;  //圆环上有10个点,编号为[0,9]

        //dp[i][j]表示走i步到点编号为j的点有多少种方法
        vector<vector<int>> dp(n + 1, vector<int>(K));

        //base case
        dp[0][0] = 1;
        for (int j = 1; j < K; ++j) {
            dp[0][j] = 0;
        }


        //状态转移
        //dp[i][j] = dp[i-1][j-1] + dp[i-1][j+1]
        //因为题目是环,这里用的数组,所以要处理j-1,j+1越界问题
        //故 dp[i][j] = dp[i-1][(j-1+K)%K] + dp[i-1][(j+1)%K]
        for (int i = 1; i <= n; ++i) {
            for (int j = 0; j < K; ++j) {
                dp[i][j] = (dp[i - 1][(j - 1 + K) % K] + dp[i - 1][(j + 1) % K] ) % mod;
            }
        }

        return dp[n][0];
    }
};
全部评论

相关推荐

点赞 评论 收藏
分享
评论
5
收藏
分享
牛客网
牛客企业服务