TZOJ 4954: 矩阵游戏

描述：

婷婷是个喜欢矩阵的小朋友,有一天她想用电脑生成一个巨大的n行m列的矩阵(你不用担心她如何存储)。她生成的这个矩阵满足一个神奇的性质：若用F[i][j]来表示矩阵中第i行第j列的元素，则F[i][j]满足下面的递推式:

F[1][1]=1
F[i,j]=a*F[i][j-1]+b (j!=1)
F[i,1]=c*F[i-1][m]+d (i!=1)
递推式中a,b,c,d都是给定的常数。

现在婷婷想知道F[n][m]的值是多少,请你帮助她。由于最终结果可能很大，你只需要输出F[n][m]除以1,000,000,007的余数。

根据上方两个式子可以进行推导出以下式子：

                ​​​​​​​        ​​​​​​​        

 矩阵快速幂相当于快速幂中的a换成了矩阵：

void mu(ll a[2][2],ll b[2][2])//矩阵乘法
{
    ll c[2][2]={0};
    for(int i=0;i<2;i++)
    {
        for(int j=0;j<2;j++)
        {
            for(int k=0;k<2;k++)
            {
                c[i][j]=(c[i][j]+a[i][k]*b[k][j])%mod;
            }
        }
    }
    for(int i=0;i<2;i++)
    for(int j=0;j<2;j++)
    a[i][j]=c[i][j];
}
 while(m)//快速幂
    {
        if(m%2)mu(C,A);
        mu(A,A);
        m=m/2;
    }//b^m

题目还有一个特别可怕的点--10^1000 000的数据范围。

 费马小定理 a^b mod p 在b很大的时候可以先用b = b % (p-1) 当然在a==1的时候还是%p;

综上所述：

 

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod=1000000007;
string s1,s2;
ll mo(string s,int x)
{
    ll d=0;
    for(int i=0;i<s.size();i++)
    {
        d=(d*10+(s[i]-'0'))%x;
    }
    return d;
}
void mu(ll a[2][2],ll b[2][2])//矩阵乘法
{
    ll c[2][2]={0};
    for(int i=0;i<2;i++)
    {
        for(int j=0;j<2;j++)
        {
            for(int k=0;k<2;k++)
            {
                c[i][j]=(c[i][j]+a[i][k]*b[k][j])%mod;
            }
        }
    }
    for(int i=0;i<2;i++)
    for(int j=0;j<2;j++)
    a[i][j]=c[i][j];
}

int main()
{
    cin>>s1>>s2;//题干中n和m非常大，采用大数
    ll a,b,c,d,n,m;
    cin>>a>>b>>c>>d;
    if(a==1)n=mo(s1,mod);else n=mo(s1,mod-1);
    if(c==1)m=mo(s2,mod);else m=mo(s2,mod-1);
    m--;n--;
    ll M[2][2],A[2][2],B[2][2],C[2][2],D[2][2];
    M[0][0]=M[0][1]=1;M[1][1]=M[1][0]=0;//a
    A[0][0]=a;A[1][0]=b;A[0][1]=0;A[1][1]=1;//b
    B[0][0]=c;B[1][0]=d;B[0][1]=0;B[1][1]=1;//c
    C[0][0]=C[1][1]=1;C[1][0]=C[0][1]=0;//sum
    while(m)//快速幂
    {
        if(m%2)mu(C,A);
        mu(A,A);
        m=m/2;
    }
    memcpy(D,C,sizeof(C));
    mu(C,B);
    while(n)
    {
        if(n%2)mu(M,C);
        mu(C,C);
        n=n/2;
    }
    mu(M,D);
    cout<<M[0][0];


}