POJ3070 - Fibonacci - 矩阵快速幂模板题

思路

矩阵快速幂模板题。敲就行。

AC代码

#include<stdio.h>
#include<iostream>
#include<string.h>
using namespace std;
const int mod=10000;

int ans[2][2],a[2][2],c[2][2];

void cheng(int a[2][2],int b[2][2])
{
    memset(c,0,sizeof(c));
    for(int i=0; i<2; i++)
    {
        for(int j=0; j<2; j++)
        {
            for(int k=0; k<2; k++)
                c[i][j]=(c[i][j]+a[i][k]*b[k][j])%mod;
        }
    }

    for(int i=0; i<2; i++)
    {
        for(int j=0; j<2; j++)
            a[i][j]=c[i][j];
    }

}
void ksm(int a[2][2],int n)
{
    memset(ans,0,sizeof(ans));
    for(int i=0; i<2; i++)
        ans[i][i]=1;
    while(n)
    {
        if(n&1)
            cheng(ans,a);
        cheng(a,a);
        n>>=1;
    }
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        if(n==-1)
            break;
        a[0][0]=a[1][0]=a[0][1]=1;
        a[1][1]=0;
        if(n==0||n==1)
            printf("%d\n",n);
        else
        {
            ksm(a,n);
            printf("%d\n",ans[0][1]);
        }
    }
    return 0;
}