矩陣快速冪模板

// poj3070
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
struct jz{
	int a[2][2];
	void init(){
		a[0][0]=a[0][1]=a[1][0]=1;
		a[1][1]=0;
	}
};
jz multi(jz a,jz b)
{
	jz c;
	for(int i=0;i<2;i++)
	{
		for(int j=0;j<2;j++)
		{
			c.a[i][j]=0;
			for(int k=0;k<2;k++)
			{
				c.a[i][j]+=(a.a[i][k]*b.a[k][j]);
			}
			c.a[i][j]%=10000;
		}
	}
	return c;
}
jz mul(jz s,int k)
{
	jz ans;
	ans.init();
	while(k)
	{
		if((k&1)!=0)
			ans=multi(ans,s);
		k>>=1;
		s= multi(s,s);  
		
	}
	return ans;
} 

int main()
{
	int n;
	while(cin>>n,~n)
	{
		if(!n)
		{
			cout<<"0"<<endl;
			continue;
		}
		jz s;
		s.init();
		s=mul(s,n-1);
		printf("%d\n",s.a[0][2]);
	}
	return 0;
}

發佈了73 篇原創文章 · 獲贊 89 · 訪問量 16萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章