1. 程式人生 > >bzoj3240: [Noi2013]矩陣遊戲

bzoj3240: [Noi2013]矩陣遊戲

printf ret content opera script clas 嵌套 strlen num

Description

婷婷是個喜歡矩陣的小朋友,有一天她想用電腦生成一個巨大的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的余數。

用 十進制快速冪 計算 一次函數的多層嵌套,時間復雜度O(logn+logm)。

#include<cstdio>
#include<cstring>
typedef long long i64;
const int P=1e9+7;
struct num{
    char a[1000077];
    int l;
    void R(){
        scanf("%s",a);
        l=strlen(a);
        int p=l-1;
        for(--a[p];a[p]<0;a[p]+=10,--a[--p]);
    }
}n,m;
struct F{
    
int a,b; F operator()(F f)const{ return (F){int(i64(a)*f.a%P),int((i64(a)*f.b+b)%P)}; } void R(){ scanf("%d%d",&a,&b); } }v1,v2; F operator^(F a,const num&n){ F v=(F){1,0},ts[10],u; ts[0]=v; for(int i=1;i<10;++i)ts[i]=a(ts[i-1]);
for(int i=0;i<n.l;++i){ u=v=v(v); u=u(u),u=u(u); v=u(v(ts[n.a[i]-0])); } return v; } int main(){ n.R();m.R(); v1.R(),v2.R(); v1=v1^m; v2=v1(v2(v1)^n); printf("%d\n",(v2.a+v2.b)%P); return 0; }

bzoj3240: [Noi2013]矩陣遊戲