【BZOJ3240】[NOI2013]矩阵游戏

题目链接

题目描述

求一个矩阵的第n行m列
递推式

$$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)\\ $$
对10^9+7 取模

n m巨大 10^1000000

Sol

先把转移到行尾的矩阵给他幂出来
然后乘上一个到下一行的转移矩阵,在幂个n-1次,然后在乘上转移到行尾的矩阵
这题就秒了

然而数据比较恶心

O(数的长度)转二进制? 不存在的

听说可以强行手动推式子用欧拉定理降幂?

推了一大堆后放弃(菜 TAT)

居然可以直接欧拉定理降次数后递推?
递推次数也能取模?(Orz)

于是就一脸懵逼地过了

原理?大概是因为最后的式子也是很多个幂相加的形式,所以就可以取模?(雾)

注意欧拉定理要求互质,所以当a或c等于1的话不能$mod\; \phi(p)$ 而要直接$mod\;p$,因为此时是线性相加的东西,这个取模的正确性还比较显然…

代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#include<cstdlib>
#include<set>
#define Set(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long ll;
const int mod=1e9+7;
struct matrix{
    int a[2][2];
    inline void clear(){Set(a,0);return;}
    inline int* operator [](int x){return a[x];}
    inline matrix operator *(matrix b){
        matrix c;c.clear();
        for(register int k=0;k<2;++k)
            for(register int i=0;i<2;++i)
                for(register int j=0;j<2;++j)
                    c[i][j]=(c[i][j]+1ll*a[i][k]*b[k][j]%mod)%mod;
        return c;
    }
}S,T1,T2;
const int N=1e6+10;
char sn[N],sm[N];
int ln,lm;
int n,m;
int a,b,c,d;
inline matrix matrix_POW(matrix A,ll k)
{
    if(k==1) return A;
    matrix c;c.clear();
    c[0][0]=c[1][1]=1;
    while(k>0){
        if(k&1) c=c*A;
        A=A*A;
        k>>=1;
    }
    return c;
}
inline void Turn(char s[],int len,int &to,int MO)
{
    to=0;
    for(register int i=1;i<=len;++i) to=(1ll*to*10+s[i]-'0')%MO;
}
int main()
{
    scanf("%s",sn+1);scanf("%s",sm+1);
    ln=strlen(sn+1);lm=strlen(sm+1);
    scanf("%d %d %d %d",&a,&b,&c,&d);
    if(a==1) Turn(sn,ln,n,mod);else Turn(sn,ln,n,mod-1);
    if(c==1) Turn(sm,lm,m,mod);else Turn(sm,lm,m,mod-1);
    T1[0][0]=a;T1[1][0]=b;T1[1][1]=1;
    T2[0][0]=c;T2[1][0]=d;T2[1][1]=1;
    S[0][0]=1;S[0][1]=1;
    T1=matrix_POW(T1,m-1);
    T2=T1*T2;T2=matrix_POW(T2,n-1);
    S=(S*T2)*T1;
    printf("%d\n",S[0][0]);
}