POJ Fibonacii （矩阵快速幂）

点击打开链接

这道题矩阵base^n 所得新的矩阵的右上角的值就是斐波那契数列第n项的值，但由于数比较大，所以要一直取模。

这道题的程序可以当作模板使用，其中矩阵是用结构体封装的，可以使代码更加简洁。

#include<cstdio>#include<iostream>#include<cstring>using namespace std;struct matrix{int a[2][2];}ans,base;const int MOD = 1e4;matrix multiply(matrix x, matrix y){    matrix tmp;    for (int i = 0; i < 2; i++)        for (int j = 0; j < 2; j++)        {            tmp.a[i][j] = 0;            for (int k = 0; k < 2; k++)                tmp.a[i][j] = (tmp.a[i][j] + x.a[i][k] * y.a[k][j]) % MOD;        }    return tmp;}int fast_mod(int n){    while (n)    {        if (n & 1) ans = multiply(ans, base);        base = multiply(base, base);        n >>= 1;    }    return ans.a[0][1];}int main(){   int n;   while (scanf("%d", &n) && n != -1)    {        ans.a[0][0] = 1; ans.a[0][1] = 0; ans.a[1][0] = 0; ans.a[1][1] = 1; // 初始化ans为单位矩阵，单位矩阵乘以任何矩阵等于那个矩阵本身        base.a[0][0] = 1; base.a[0][1] = 1; base.a[1][0] = 1; base.a[1][1] = 0;        printf("%d\n", fast_mod(n));    }    return 0;}