九度OJ 1081 递推数列 -- 矩阵二分乘法

题目地址：http://ac.jobdu.com/problem.php?pid=1081

 

题目描述：

给定a0,a1,以及an=p*a(n-1) + q*a(n-2)中的p,q。这里n >= 2。 求第k个数对10000的模。

输入：

输入包括5个整数：a0、a1、p、q、k。

输出：

第k个数a(k)对10000的模。

样例输入：

20 1 1 14 5

样例输出：

8359

#include <stdio.h>
 
#define MOD 10000
 
typedef struct matrix{
    int a00, a01, a10, a11;
}Matrix;
 
void MatrixMul(Matrix * m, Matrix * n){
    Matrix tmp;
    tmp.a00 = (m->a00 * n->a00 + m->a01 * n->a10) % MOD;
    tmp.a01 = (m->a00 * n->a01 + m->a01 * n->a11) % MOD;
    tmp.a10 = (m->a10 * n->a00 + m->a11 * n->a10) % MOD;
    tmp.a11 = (m->a10 * n->a01 + m->a11 * n->a11) % MOD;
    *m = tmp;
}
 
int main(void){
    int a0, a1, p, q, k;
 
    while (scanf ("%d%d%d%d%d",
            &a0, &a1, &p, &q, &k) != EOF){
        if (k == 0) {printf ("%d\n", a0 % MOD); continue;}
        if (k == 1) {printf ("%d\n", a1 % MOD); continue;}
        Matrix pq = {p%MOD, q%MOD, 1, 0};
        Matrix ans = {1, 0, 0, 1};
        --k;
        while (k >= 1){
            if ((k & 1) == 1)
                MatrixMul (&ans, &pq);
            MatrixMul (&pq, &pq);
            k = k >> 1;
        }
        printf ("%d\n", (ans.a00*a1+ans.a01*a0) % MOD);
    }
 
    return 0;
}

 

参考资料：十个利用矩阵乘法解决的经典题目