斐波那契数列

斐波那契数列是由如下递推式定义的数列:

F[0]=0，F[1]=1，F[n+2]=F[n+1]+F[n]。

求这个数列第n项的值对10^4取余后的结果。

限制条件(0<=n<=10^16)(注意n可以非常非常大。。。)

代码如下:

1.Matrix结构体

#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
const int N=100+10;
const int M=10000;
struct Matrix
{
                int row,col;
                ll mat[N][N];
                void init(int row,int col,bool one=false){
                      this->row=row;
                      this->col=col;
                      memset(mat,0,sizeof(mat));
                      if(!one) return;
                      for(int i=1;i<row;i++){
                                mat[i][i]=1;
                      }
                }
                Matrix operator * (const Matrix& B){
                                Matrix C;
                                C.init(row,B.col);
                                for(int k=0;k<col;k++){
                                                for(int i=0;i<row;i++){
                                                                if(mat[i][k]==0) continue;
                                                                for(int j=0;j<B.col;j++){
                                                                                if(B.mat[k][j]==0) continue;
                                                                                C.mat[i][j]+=(mat[i][k]*B.mat[k][j])%M;
                                                                                C.mat[i][j]=C.mat[i][j]%M;
                                                                }
                                                }
                                }
                                return C;
                }
                Matrix pow(ll n){
                                Matrix B,A=*this;
                                B.init(row,col,1);
                                while(n>0){
                                                if(n&1) B=B*A;
                                                A=A*A;
                                                n>>=1;
                                }
                                return B;
                }
};
int main()
{
       int n;
       while(cin>>n){
                                Matrix A;
                                A.init(2,2);
                                A.mat[0][0]=1;A.mat[0][1]=1;
                                A.mat[1][0]=1;A.mat[1][1]=0;
                                A=A.pow(n);
                                cout<<A.mat[1][0]<<endl;
       }
}

2.vector二维数组

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
const double EPS= 1e-8 ;
typedef vector<int> vec;
typedef vector<vec> mat;
typedef long long ll;
const int M=10000;
//计算A*B
mat mul(mat& A,mat& B)
{
    mat C(A.size(),vec(B[0].size()));
    for(int i=0;i<A.size();i++){
        for(int k=0;k<C.size();k++){
            for(int j=0;j<B.size();j++){
                C[i][j]=(C[i][j]+A[i][k]*B[k][j])%M;
            }
        }
    }
    return C;
}
//计算A^n
mat pow(mat A,ll n)
{
    mat B(A.size(),vec(A.size()));
    for(int i=0;i<A.size();i++){
        B[i][i]=1;
    }
    while(n>0){
        if(n&1) B=mul(B,A);
        A=mul(A,A);
        n>>=1;
    }
    return B;
}
int main()
{
    ll n;
    cin>>n;
    mat A(2,vec(2));
    A[0][0]=1;A[0][1]=1;
    A[1][0]=1;A[1][1]=0;
    A=pow(A,n);
    cout<<A[1][0]<<endl;
    return 0;
}