UVA 10870 递推关系 矩阵快速幂

题意：f(n) = a1 ＊f(n - 1) + a2 ＊f(n - 2) + a3 ＊f(n - 3) + … + ad＊ f(n - d),  n > d.求f(n)

见白书 155页。

 由于n太大，不能直接递推，需要用矩阵快速幂来解决，时间复杂度为O(d^3logn)
    举例，d=5的矩阵关系式为：
                |a1 a2 a3 a4 a5|                 | f[n]     |      | f[n+1] |  
                |1                      |                 | f[n-1]  |      | f[n]    |  
                |     1                 |  *              | f[n-2]  | =   | f[n-1] | (空白处为0)
                |          1            |                 | f[n-3]  |      | f[n-2] |
                |              1        |                 | f[n-4]  |      | f[n-3] |

F(n)=A*F(n-1);

--->> F(n)=A^(n-d)*F(d);

#include<bits/stdc++.h>
#define LL long long
#define INF 0x3f3f3f3f
#define mem(a,b) memset(a,b,sizeof(a))
#define bug puts("***********")
using namespace std;
const int N =20;
struct Mat{
    LL mat[N][N];
}sta,fi;
LL n,m;
int d;
Mat mul(Mat a,Mat b){
    Mat tmp;
    mem(tmp.mat,0);
    for(int i=1;i<=d;i++){  
        for(int j=1;j<=d;j++){
            for(int k=1;k<=d;k++){
                tmp.mat[i][j]=(tmp.mat[i][j]+a.mat[i][k]*b.mat[k][j]%m)%m;
            }
        }
    }
    return tmp;
}
Mat quick(LL k){
    Mat tmp;
    mem(tmp.mat,0);
    for(int i=0;i<=20;i++)      ///初始化为单位矩阵
        tmp.mat[i][i]=1;
    while(k){
        if(k&1) tmp=mul(tmp,sta);
        k>>=1;
        sta=mul(sta,sta);
    }
    return tmp;
}
int main(){
    while(~scanf("%d%lld%lld",&d,&n,&m)){
        if(!n&&!m&&!n)break;
        mem(sta.mat,0);
        mem(fi.mat,0);
        for(int i=2;i<=d;i++){
            sta.mat[i][i-1]=1;
        }
        for(int i=1;i<=d;i++){
            scanf("%lld",&sta.mat[1][i]);
            sta.mat[1][i]%=m;
        }
        for(int i=1;i<=d;i++){
            scanf("%lld",&fi.mat[d-i+1][1]);
            fi.mat[d-i+1][1]%=m;
        }
        if(n<=(LL)d){
            printf("%lld\n",fi.mat[d-n+1][1]);
            continue;
        }
        fi=mul(quick(n-d),fi);     ///方向别反了
        printf("%lld\n",fi.mat[1][1]);
    }
    return 0;
}