斐波那契第k項

按自己風格寫的,以後就直接用了。

#include<iostream>
#include<cstdio>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;

//    [fn]    =  [1,1]  *  [fn-1]
//    [fn-1]     [1,0]     [fn-2]
//   以0作爲第一項 

ll m[3][3],g[3][3];

ll fgfk(int n) {
	m[1][1]=1,m[1][2]=1,m[2][1]=1,m[2][2]=0;
	g[1][1]=1,g[1][2]=0,g[2][1]=0,g[2][2]=1;
	if(n==1) {
		return 0;
	}else if(n==2) {
		return 1;
	}
	n-=2;
	while(n>0) {
		if(n&1) {
			ll a1=(g[1][1]*m[1][1]+g[1][2]*m[2][1])%mod;
		    ll a2=(g[1][1]*m[1][2]+g[1][2]*m[2][2])%mod;
		    ll a3=(g[2][1]*m[1][1]+g[2][2]*m[2][1])%mod;
		    ll a4=(g[2][1]*m[1][2]+g[2][2]*m[2][2])%mod;
		    g[1][1]=a1,g[1][2]=a2,g[2][1]=a3,g[2][2]=a4;
		}
		ll a1=(m[1][1]*m[1][1]+m[1][2]*m[2][1])%mod;
		ll a2=(m[1][1]*m[1][2]+m[1][2]*m[2][2])%mod;
		ll a3=(m[2][1]*m[1][1]+m[2][2]*m[2][1])%mod;
		ll a4=(m[2][1]*m[1][2]+m[2][2]*m[2][2])%mod;
		m[1][1]=a1,m[1][2]=a2,m[2][1]=a3,m[2][2]=a4;
		n>>=1;
	}
	return g[1][1];
}

int main() {
	int n;
	while(scanf("%d",&n)==1) {
		printf("第%d項 = %lld\n",n,fgfk(n));
	}
} 


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章