牛客巅峰赛--数列求值(矩阵快速幂)

题目链接:https://ac.nowcoder.com/acm/contest/6631/C

  • 直接矩阵快速幂就可。
  • 基础矩阵base=\begin{bmatrix} b&c \\ 1&0 \end{bmatrix},
  • 那么\begin{bmatrix} a_n\\a_{n-1} \end{bmatrix} = \begin{bmatrix} b&c \\ 1&0 \end{bmatrix}^{n-1}* \begin{bmatrix} a1\\a0 \end{bmatrix}

AC代码: 

typedef long long LL;
const int mod = 1e9+7;
class Solution {
public:
    struct mat{
        LL m[2][2];
    };
    
    mat muti(mat a,mat b)//矩阵乘法
    {
        mat tmp;
        for(int i=0;i<2;i++)
        {
            for(int j=0;j<2;j++)
            {
                tmp.m[i][j]=0;
                for(int k=0;k<2;k++)
                {
                    tmp.m[i][j]+=a.m[i][k]*b.m[k][j];
                    tmp.m[i][j]%=mod;
                }
            }
        }
        return tmp;
    }
    LL mat_pow(mat a,LL n)//矩阵快速幂
    {
        mat ans={1,0,0,1};
        while(n)
        {
            if(n&1)ans=muti(ans,a);
            a=muti(a,a);
            n>>=1;
        }
        return ans.m[0][0];
    }
    long long nthElement(long long n, long long b, long long c) {
        if(n==0)return 0;
        if(n==1)return 1;
        mat base={b,c,1,0};
        return mat_pow(base, n-1);
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43911947/article/details/107712249