描述
婷婷是个喜欢矩阵的小朋友,有一天她想用电脑生成一个巨大的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
题解:
由题目中的第一个表达式 F[i,j] = a * F[ i ][ j-1 ] + b ( j!=1 ) 可以轻易推出矩阵中每一行的递推关系
而第二个表达式也可以推出每一行最后一项和下一行第一项之间的关系。
将公式不停带入,可以得到:
F(n,m)=a·F(n,m-1)+b=(a^(m-1))F(n,1)+b+ab+...+(a^(m-2))b=(a^(m-1))c·F(n-1,m)+(a^(m-1))d+b+ab+...+(a^(m-2))b
记p=b+ab+...+(a^(m-2))b, x=(a^(m-1))c, y=(a^(m-1))d+p, q=y+xy+...+(x^(n-2))y
那么F(n,m)=x·F(n-1,m)+y=(x^(n-1))F(1,m)+q=(x^(n-1))(a^(m-1)+p)+q;
但是 n 和 m 巨大的数据范围(10^1000000)所以要压缩n和m,将 n 和 m 取模,需要用到费马小定理,即 F(n,m) = F ( n%(mod-1) , m%(mod-1) ) 。让后再用矩阵快速幂进一步加速处理。
#include<iostream>
#include<stack>
#include<cmath>
#include<iomanip>
#include<string.h>
#include<string>
#pragma warning(disable : 4996)
using namespace std;
#define MOD 1000000007
string n, m;
long long N,M,a, b, c, d;
long long A[2][2], B[2][2], C[2][2],D[2][2],DD[2][2];
long long toLongLong(string x,long long mod)
{
long long l = x.size();
long long ans = 0;
for (int i = 0; i < l; i++)
{
ans = (ans * 10 + (x[i] - '0')) % mod;
}
return ans;
}//将n和m降低次数转化为long long
void mul(long long x[2][2], long long y[2][2])
{
long long res[2][2] = { 0 };
for (int i = 0; i < 2; i++)
{
for (int j = 0; j < 2; j++)
{
for (int k = 0; k < 2; k++)
res[i][j] = (res[i][j] + x[i][k] * y[k][j]) % MOD;
}
}
for (int i = 0; i < 2; i++)
{
for (int j = 0; j < 2; j++) x[i][j] = res[i][j];
}
}
//矩阵乘法
int main()
{
//加快cin和cout的速度。
ios::sync_with_stdio(false);
cin.tie(0);
cout.tie(0);
//原始数据输入
cin >> n >> m >> a >> b >> c >> d;
//处理n和m的大小,可以通过费马小定理得到等效但较小的值,当a=1和c=1时特殊处理。
if (a == 1) N = toLongLong(n,MOD);
else N = toLongLong(n, MOD - 1);
if (c == 1)M = toLongLong(m, MOD);
else M = toLongLong(m, MOD - 1);
M--;
N--;
A[0][0] = 1;
A[0][1] = 1;
A[1][1] = 0;
A[1][0] = 0;
B[0][0] = a;
B[1][0] = b;
B[0][1] = 0;
B[1][1] = 1;
C[0][0] = c;
C[1][0] = d;
C[0][1] = 0;
C[1][1] = 1;
D[0][0] = D[1][1] = 1;
D[0][1] = D[1][0] = 0;
while (M)//快速幂
{
if (M & 1) mul(D, B);
mul(B, B);
M >>= 1;
}
DD[0][0] = D[0][0];
DD[0][1] = D[0][1];
DD[1][0] = D[1][0];
DD[1][1] = D[1][1];
mul(D, C);
while (N)//快速幂
{
if (N & 1) mul(A, D);
mul(D, D);
N >>= 1;
}
mul(A, DD);
cout << A[0][0] << endl;
return 0;
}