描述
婷婷是个喜欢矩阵的小朋友,有一天她想用电脑生成一个巨大的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的余数。
输入
一行有六个整数n,m,a,b,c,d。意义如题所述。
1<=N,M<=10^1000 000,a<=a,b,c,d<=10^9
输出
包含一个整数,表示F[n][m]除以1,000,000,007的余数
样例输入
3 4 1 3 2 6
样例输出
85
提示
样例中的矩阵为:
1 4 7 10
26 29 32 35
76 79 82 85
首先 由于N,M<=10^1000000,我们需要用到费马定理才能使代码不超时
然后就是快速幂和矩阵乘法的结合
以下为ac代码
#include<bits/stdc++.h>
using namespace std;
const int q=1e9+7;
typedef long long ll;
ll aa[2][2],bb[2][2],cc[2][2],dd[2][2],mi[2][2],aa1[2][2];
string N,M;
void km(ll x[][2],ll y[][2])
{
for(int i=0;i<2;i++)
{
for(int j=0;j<2;j++)
{
ll s=0;
for(int k=0;k<2;k++)
{
s=(s+x[i][k]*y[k][j])%q;
}
mi[i][j]=s;
}
}
for(int i=0;i<2;i++)
{
for(int j=0;j<2;j++) x[i][j]=mi[i][j];
}
}
ll getmo(string x,ll k)
{
ll p=0;
for(ll i=0;i<x.size();i++) p=(p*10+(x[i]-'0'))%k;
return p;
}
int main()
{
ll a,b,c,d,n,m;
cin>>N>>M>>a>>b>>c>>d;
if(a==1) n=getmo(N,q);
else n=getmo(N,q-1);
if(c==1) m=getmo(M,q);
else m=getmo(M,q-1);
n-=1,m-=1;
aa[0][0]=1,aa[0][1]=0,aa[1][0]=0,aa[1][1]=1;
bb[0][0]=a,bb[0][1]=0,bb[1][0]=b,bb[1][1]=1;
cc[0][0]=c,cc[0][1]=0,cc[1][0]=d,cc[1][1]=1;
dd[0][0]=1,dd[0][1]=1,dd[1][0]=0,dd[1][1]=0;
while(m)
{
if(m&1) km(aa,bb);
km(bb,bb);
m>>=1;
}
aa1[0][0]=aa[0][0],aa1[0][1]=aa[0][1],aa1[1][0]=aa[1][0],aa1[1][1]=aa[1][1];
km(aa,cc);
while(n)
{
if(n&1) km(dd,aa);
km(aa,aa);
n>>=1;
}
km(dd,aa1);
cout<<dd[0][0];
}