快速冪

第一次寫快速冪,學會了肥波數列的快速算法,矩陣是個好東西
 
 
 
 
 
 
/**
 *
 *Dtcg
 *2014.12.05
 ***************/
#include <cmath>
#include <vector>
#include <sstream>
#include <cstring>
#include <fstream>
#include <iostream>
#include <algorithm>
using namespace std;
int _2Fi[30][2][2];
vector<int >s;

inline void matmul2(int x){
	for (int t=0; t<2; t++)
		for (int i=0; i<2; i++)
			for (int j=0; j<2; j++)
				_2Fi[x][t][i] += _2Fi[x-1][t][j] * _2Fi[x-1][i][j] %10000;
}

inline void matmul(int p[2][2],int k[2][2],int u[2][2]){
	for (int t=0;t<2;t++)
		for (int i=0;i<2;i++)
			for (int j=0;j<2;j++)
				p[t][i]+=k[t][j]*u[i][j]%10000;
}

int main(){
	memset(_2Fi,0,sizeof _2Fi);
	_2Fi[0][0][1]=_2Fi[0][0][0]=_2Fi[0][1][0]=1;
	for(int i=1;i<30;i++)
		matmul2(i);
	int n;
	while(cin>>n,n>=0){
		int ans [2][2][2]={{{1,0},{0,1}}};
		s.clear();
		for(int j=0;j<32;j++)
			if((n-1)&(1<<j))
				s.push_back(j);
		bool pre=0,now=1;
		for(int i=0;i<s.size();i++){
			memset(ans[now],0,sizeof(ans[now]));
			matmul(ans[now], ans[pre],_2Fi[s[i]]);
			swap(pre,now);
		}
		cout<<ans[pre][0][0]%10000<<endl;
	}
	return 0;
}

你可能感兴趣的:(快速冪)