935. 骑士拨号器--c++解题方法二

935. 骑士拨号器

思路：将0-9分为四个部分：{1,3,7,9},{4,6},{2,8},{0}

代码

#include
#include
#include

using namespace std;

//骑士跳“日”
long M = 1e9 + 7;

//{1,3,7,9},{4,6},{2,8},{0}
int knightDialer(int N) {
    if(N == 1){
        return 10;
    }

    vector<long> dp1(4);
    //{1,3,7,9}有四个数
    dp1[0] = 4;
    //{4，6}有两个数
    dp1[1] = 2;
    //{2，8}有两个数
    dp1[2] = 2;
    //{0}有一个数
    dp1[3] = 1;
    vector<long> dp2(4);

    int step =1;
    while(step<N){
        //{1,3,7,9}可以由{4,6}和{2,8}得到，由于下一步只能选特定一个数，
        //但是实际可选的有两个，所以需要左移1位
        dp2[0] = ((dp1[1] + dp1[2])<<1)%M;
        //{4,6}可以由{1,3,7,9}移动一步得到，或者由{0}移动一步得到，
        //但是由{0}出发，有两种选择，所以需要左移1位
        dp2[1] = (dp1[0] + (dp1[3]<<1))%M;
        //{2，8}可以由{1,3,7,9}移动一步得到
        dp2[2] = (dp1[0])%M;
        //{0}可以由{4，6}移动一步得到
        dp2[3] = (dp1[1])%M;

        dp1 = dp2;
        step++;
    }

    long result = 0;
    for(int i=0;i<dp1.size();i++){
        result += dp1[i];
    }
    return result%M;
    
}
int main(){
    int resultValue = knightDialer(161);
    cout<<resultValue<<endl;
    return 0;
}