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的余数。

输入：

一行有六个整数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的余数

题意：让我们算出f[n,m]通过上面的递推式，但n,m的数据范围很大，有10^1000 000。

思路：

先通过上面的递推式推出可以快速求出答案的公式，由于数据量庞大直接这样是不行的

推导过程大致如下：

当a不等于1时：

f[n+1,1]=A^n*f[1,1]+B*(A^n-1)/(A-1)   

当a等于1时：

f[n+1,1]=f[1,1]+n*D   (c=1)

f[n+1,1]=c^n(f[1,1])+D*(c^n-1)/(c-1)

所以我们可以通过这样快速求出f[n+1,1]，然后带入题目中所给的第三个例子，直接先减d再除c可以得到f[n,m]

因为n,m很大，所以当我们要进行优化，首先当n,m为系数时是可以直接在乘的时候进行取模，而作为指数的时候，则欧拉降幂进行优化，要值得注意的是，在进行除法的时候都要取逆元进行计算。

欧拉降幂：

 因为题目给出的模数是质数       φ(p)=p−1 (对于质数p)

代码：

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL)
#define x first
#define y second
using namespace std;
typedef long long ll;
const int N=1000010,mod=1000000007;
char strn[N],strm[N];
ll a,b,c,d;
ll n,m,nn,mm; 
ll fn1;
ll quick_mi(ll a,ll b)
{
	ll res=1;
	a=a%mod;
	while(b)
	{
		if(b&1) res=(res*a)%mod;
		a=(a*a)%mod;
		b>>=1;
	}
	return res%mod;
}
ll ni(ll x)
{
	return quick_mi(x,mod-2);
}
void init()
{
	int len=strlen(strn+1);
	int lem=strlen(strm+1);
	//n,m是作为系数时对mod取模后的数字,nn,mm是作为指数时对mod取模的数字； 
	for(int i=1;i<=len;i++)
	{
		n=(n*10+strn[i]-'0')%mod;
		nn=(nn*10+strn[i]-'0')%(mod-1);
	}
	for(int i=1;i<=lem;i++)
	{
		m=(m*10+strm[i]-'0')%mod;
		mm=(mm*10+strm[i]-'0')%(mod-1);
	}
}
void solve()
{
	cin>>strn+1>>strm+1;
	cin>>a>>b>>c>>d;
	init();//处理一下n,m； 
	//求出推导式中的A，B，D；
	//分情况：
	if(a!=1)
	{
		ll aa=quick_mi(a,(mm-1+mod)%mod)%mod;
		ll A=(aa*c)%mod;
		ll B=(((c*b)%mod*(aa-1+mod)%mod*ni(a-1+mod)%mod)%mod+d)%mod;
		ll bb=quick_mi(A,nn);
		fn1=(bb+((B*(bb-1+mod)%mod)*ni(A-1+mod)%mod))%mod;
	} 
	else
	{
		ll D=(((b*c)%mod*(m-1+mod))%mod+d)%mod;
		if(c==1)
		{
			fn1=(1+n*D%mod)%mod;
		}
		else
		{
			ll cc=quick_mi(c,nn);
			fn1=(cc+D*(cc-1+mod)%mod*ni(c-1+mod)%mod)%mod;
		}
	}
	printf("%lld\n",((fn1-d+mod)*ni(c)%mod+mod)%mod);
}
int main()
{
	//IOS;
	int T;
	T=1;
	while(T--)
	{
		solve();
	}
}