POJ 3070 Fibonacci（矩阵快速幂）

题目链接

题意 ： 用矩阵相乘求斐波那契数的后四位。

思路 ：基本上纯矩阵快速幂。

//3070
#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

struct Matrix
{
    int v[2][2];
};
int n;

Matrix matrix_mul(Matrix a,Matrix b)
{
    Matrix c;
    for(int i = 0 ; i < 2 ; i++)
    {
        for(int j = 0 ; j < 2 ; j++)
        {
            c.v[i][j] = 0 ;
            for(int k = 0 ; k < 2 ; k++)
                c.v[i][j]=(c.v[i][j]+a.v[i][k]*b.v[k][j])%10000;
        }
    }
    return c;
}

int matrix_mi()
{
        Matrix p,t ;
        p.v[0][0] = p.v[0][1] = p.v[1][0] = 1 ;
        p.v[1][1] = 0;
        t.v[0][0] = t.v[1][1] = 1 ;//t是单位向量
        t.v[0][1] = t.v[1][0] = 0 ;
        while(n)
        {
            if(n&1)//奇数
                t = matrix_mul(t,p);
            n = n>>1 ;
            p = matrix_mul(p,p);
        }
        return t.v[1][0] ;
}
int main()
{
    while(scanf("%d",&n)!=EOF&&n!=-1)
    {
        if(n == 0 || n == 1)
        {
            cout<<n<<endl;
            continue;
        }
        int ans = matrix_mi();
        cout<<ans<<endl;
    }
    return 0;
}