TZOJ----矩阵游戏

题意

婷婷是个喜欢矩阵的小朋友,有一天她想用电脑生成一个巨大的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 

 

 

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5,mod1=1e9+7, mod2=mod1-1;
typedef long long ll;
#define int ll

int a,b,c,d,nu,nv,mu,mv;

void init(int &x,int &y){
    string s;
    cin>>s;
    for(int i=0;i<s.length();i++){
        x=(x*10%mod1+s[i]-'0') % mod1;
		y=(y*10%mod2+s[i]-'0') % mod2;
    }
}

int ksm(int a ,int b){
    int ans=1;
    while(b){
        if(b&1)	ans=ans*a%mod1;
		a=a*a%mod1;
		b/=2;
    }
    return ans;
}

int fun(int x, int y){
	int ans;
	if(x==1) ans=(1+nu*y%mod1)%mod1;
	else{
		ans=ksm(x,nv);
		ans=(ans+y*(ksm(x, nv)-1+mod1)%mod1*ksm(x-1, mod1-2)%mod1)%mod1;
	}
	return ans;
}


signed main(){
    init(nu, nv);
	init(mu, mv);
	cin>>a>>b>>c>>d;
	int ans;
	if(a==1){
		int dd=c*b%mod1*(mu-1+mod1)%mod1;
		dd=(dd+d)%mod1;
		ans = fun(c, dd);
	}
	else{
		int aa=c*ksm(a,(mv-1+mod2)%mod2+mod2)%mod1;
		int bb=(c*b%mod1*(ksm(a, (mv-1+mod2)%mod2+mod2)-1+mod1)%mod1*ksm(a-1, mod1-2)%mod1+d)%mod1;
		ans = fun(aa, bb);
	}
	ans=(ans-d+mod1)%mod1;
	ans=ans*ksm(c, mod1-2)%mod1;
	cout<<ans<<endl;
}