码迷,mamicode.com
首页 > 其他好文 > 详细

矩阵乘法加速fib数列

时间:2017-08-15 22:54:28      阅读:199      评论:0      收藏:0      [点我收藏+]

标签:span   clear   str   struct   print   ring   mod   efi   his   

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
#define LL long long
const int mod = 1e9+7;
const int N = 2;
struct Matrix{
    int a[N][N];
    Matrix(){
        this -> clear();
    }
    void clear(){
        memset(a,0,sizeof(a));
    }
    void setone(){
        this ->clear();for(int i=0;i<N;++i)
            a[i][i]=1;
    }
    Matrix operator * (const Matrix &x) {
         Matrix c;
         for (int k=0;k<N;++k)
             for (int i=0;i<N;++i)
                 for (int j=0;j<N;++j)
                    c.a[i][j]=(c.a[i][j]+(LL)a[i][k]*x.a[k][j])%mod;
    }   
};
int fibn(int x)
{
    Matrix a,b;
    a.a[0][0]=a.a[0][1]=a.a[1][0]=1;
    b.setone();
    while(x)
    {
        if(x&1) b=b*a;
        a=a*a;
        x>>=1;
    }
    return (b.a[0][0]+b.a[0][1])%mod;
}   
int main()
{
    int n;
    while (~scanf("%d",&n))
    {
        int (n==1||n==2)puts("1");
        else printf("%d\n",fibn(n-2));
    }
    return 0;
}

 

矩阵乘法加速fib数列

标签:span   clear   str   struct   print   ring   mod   efi   his   

原文地址:http://www.cnblogs.com/sssy/p/7368034.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!