题解 | #圆环回原点#

圆环回原点

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];
    }
};
全部评论

相关推荐

小红书 后端选手 n*16*1.18+签字费期权
点赞 评论 收藏
分享
ArisRobert:统一解释一下,第4点的意思是,公司按需通知员工,没被通知到的员工是没法去上班的,所以只要没被通知到,就自动离职。就是一种比较抽象的裁员。
点赞 评论 收藏
分享
10-09 00:50
已编辑
长江大学 算法工程师
不期而遇的夏天:1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
评论
5
收藏
分享
牛客网
牛客企业服务