矩阵游戏
描述
婷婷是个喜欢矩阵的小朋友,有一天她想用电脑生成一个巨大的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的余数
题解
m,n的数据范围在10^1000000,必须要使用方法加速
我使用的是矩阵;
用ma储存f[1,1];
ma=,A=
,B=
F[i,j]=a*F[i][j-1]+b (j!=1)推的是F[i,1]->F[i,m]
要推n-1次
从F[1,1]->F[1,2]:
ma*A=
从F[1,1]->F[1,3]:
ma*A*A=
则F[1,1]->F[1,m] ma=ma*A^m-1;
F[i,1]=c*F[i-1][m]+d (i!=1)推的是F[i,m]->F[i+1,1];
从F[i,m]->F[i+1,1],ma=ma*A^m-1*B
要推n-1次
ma=(ma*A^m-1*B)^n-1;
最后从F[n,1]->F[n,m]:
ma=(ma*A^m-1*B)^n-1*A^m-1;
因为n,m太大,可以使用欧拉降幂来减少次数,
1e+7为质数,则a不为1,成立
#include<bits/stdc++.h>
using namespace std;
int mod=1000000007;
typedef long long ll;
string N,M;
ll a,b,c,d;
ll ma[2][2]={1,1,0,0},A[2][2],B[2][2],C[2][2];
//C用于储存 A^m-1;
ll f(string x,int m)
{
ll l=x.size();
ll sum=0;
for(int i=0;i<l;i++)
{
sum=(sum*10+(x[i]-'0'))%m;
}
return sum;
}
void f1(ll x[2][2],ll y[2][2])
{
ll D[2][2]={0,0,0,0};
for(int i=0;i<=1;i++)
{
for(int j=0;j<=1;j++)
{
for(int k=0;k<=1;k++)
{
D[i][j]=(D[i][j]+x[i][k]*y[k][j])%mod;
}
}
}
for(int i=0;i<=1;i++)
{
for(int j=0;j<=1;j++)
{
x[i][j]= D[i][j];
}
}
}
int main()
{
ll n,m,D[2][2]={1,0,0,1};//D用于运算
cin>>N>>M>>a>>b>>c>>d;
//cout<<N<<" "<<M<<" "<<a<<" "<<b<<" "<<c<<" "<<d<<endl;
if(a==1) n=f(N,mod);
else n=f(N,mod-1);
if(c==1) m=f(M,mod);
else m=f(M,mod-1);
n--;
m--;
A[0][0]=a,A[0][1]=0,A[1][0]=b,A[1][1]=1;
B[0][0]=c,B[0][1]=0,B[1][0]=d,B[1][1]=1;
D[0][0]=D[1][1]=1;D[0][1]=D[1][0]=0;
while(m)
{
if(m&1) f1(D,A);
f1(A,A);
m/=2;
}
for(int i=0;i<=1;i++)
{
for(int j=0;j<=1;j++)
{
C[i][j]= D[i][j];
}
}
//C=A^m-1;
f1(D,B);
//D=A^m-1*B;
while(n)
{
if(n&1) f1(ma,D);
f1(D,D);
n/=2;
}
f1(ma,C);
printf("%lld",ma[0][0]);
return 0;
}