矩阵快速幂 模板

矩阵快速幂模板

struct matrix
{
    long long m[3][3];
    matrix()
    {
        memset(m,0,sizeof(m));
    }
    matrix operator*(const matrix b)const
    {
        long long i,j,k;
        matrix ans;
        for(i=0; i<3; i++)
        {
            for(j=0; j<3; j++)
            {
                ans.m[i][j]=0;
                for(k=0; k<3; k++)
                {
                    ans.m[i][j]+=m[i][k]*b.m[k][j];
                    ans.m[i][j]%=mod;
                }
            }
        }
        return ans;
    }
    void set(long long a,long long b)
    {
        m[0][0]=a;
        m[0][1]=b;
        m[1][0]=1;
        m[1][1]=0;
        m[2][0]=1;
        m[2][1]=0;
        m[2][2]=1;
    }
} e;
long long qpow(long long n,long long f1,long long f2,long long a,long long b)
{
    n--;
    e.m[0][0]=1;
    e.m[1][1]=1;
    e.m[2][2]=1;
    matrix temp,ans;
    ans=e;
    temp.set(a,b);
    while(n)
    {
        if(n&1)
            ans=ans*temp;
        temp=temp*temp;
        n>>=1;
    }
    long long fn=(ans.m[1][1]*f1+ans.m[1][0]*f2)%mod;
    long long sum=(ans.m[2][0]*f2+ans.m[2][1]*f1+ans.m[2][2]*f1)%mod;
    return fn;
}