poj 3070 Fibonacci 矩阵快速幂

来源:互联网 发布:数据分析材料 编辑:程序博客网 时间:2024/06/06 00:44

点击打开链接

题意求:F[n]%mod n<=1e9 F[i] 为fibonacci项 , 显然线性递推时间空间都不行.

由矩阵乘法

得到F[i]第n项为矩阵[[1,1] [1,0]]的n次幂的a[1][2]项,利用快速求出矩阵即可得到F[n]

#include <iostream>#include <algorithm>#include <cstring>#include <map>#include <vector>using namespace std;typedef long long ll;const int mod=1e4;const int N=2e5+20;struct matrix{int a[3][3];void init(){a[1][1]=1;a[1][2]=a[2][1]=1;a[2][2]=0;}};matrix product(matrix a,matrix b)//矩阵乘法 {matrix c;for(int i=1;i<=2;i++){for(int j=1;j<=2;j++){c.a[i][j]=0;for(int k=1;k<=2;k++){c.a[i][j]+=(a.a[i][k]*b.a[k][j]);}c.a[i][j]%=mod;}}return c;}matrix mul(matrix s,int k){matrix ans;ans.init();while(k){if(k%2)ans=product(ans,s);s=product(s,s);k=k/2;}return ans;}int n;int main(){while(cin>>n&&n!=-1){if(!n){cout<<0<<endl;continue;}matrix s;s.init();s=mul(s,n-1);cout<<s.a[1][2]%mod<<endl;}return 0;} 


0 0
原创粉丝点击