A Simple Math Problem HDU 1757

很简单的快速幂模版，当n<10时输出n，当n>=10斐波那契，找出他的邻接矩阵就行了

​
#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

long long n,mod;

struct node{
	int m[10][10];
}ans,t;

node operator *(node a,node b)
{
	node res;
	for(int i=0;i<10;i++)
		for(int j=0;j<10;j++)
		{
			res.m[i][j]=0;
			for(int k=0;k<10;k++)
				res.m[i][j]+=(a.m[i][k]*b.m[k][j])%mod;
			res.m[i][j]%=mod;
		}
	return res;
}

void slove()
{
	for(int i=0;i<10;i++)
		for(int j=0;j<10;j++)
			ans.m[i][j]=0;
	for(int i=0;i<10;i++)
		ans.m[i][i]=1;
	while(n)
	{
		if(n&1)
			ans=t*ans;
		t=t*t;
		n/=2;
	}
	long long sum=0;
	for(int i=0;i<10;i++)
	{
		sum+=(9-i)*ans.m[0][i]%mod;
		sum%mod;
	}	
	cout<<sum%mod<<endl;
}

int main()
{
	while(cin>>n>>mod)
	{
		if(n<10)
		{
			cout<<n<<endl;
			continue;
		}
		else
		{
			for(int i=0;i<10;i++)
				for(int j=0;j<10;j++)
					t.m[i][j]=0;
			for(int i=0;i<10;i++)
				cin>>t.m[0][i];
			for(int i=0;i<10;i++)
				t.m[i+1][i]=1;
			n-=9;
			slove();
		}
	}
	return 0;
}

​

有一小部分换了也可以

#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

long long n,mod;

struct node{
	int m[10][10];
}ans,t;

node operator *(node a,node b)
{
	node res;
	for(int i=0;i<10;i++)
		for(int j=0;j<10;j++)
		{
			res.m[i][j]=0;
			for(int k=0;k<10;k++)
				res.m[i][j]+=(a.m[i][k]*b.m[k][j])%mod;
			res.m[i][j]%=mod;
		}
	return res;
}

void slove()
{
	for(int i=0;i<10;i++)
		for(int j=0;j<10;j++)
			ans.m[i][j]=0;
	for(int i=0;i<10;i++)
		ans.m[i][0]=9-i;
	while(n)
	{
		if(n&1)
			ans=t*ans;
		t=t*t;
		n/=2;
	}
	long long sum=0;
	/*
	for(int i=0;i<10;i++)
	{
		sum+=(9-i)*ans.m[0][i]%mod;
		sum%mod;
	}
	*/
	cout<<ans.m[0][0]%mod<<endl;
}

int main()
{
	while(cin>>n>>mod)
	{
		if(n<10)
		{
			cout<<n<<endl;
			continue;
		}
		else
		{
			for(int i=0;i<10;i++)
				for(int j=0;j<10;j++)
					t.m[i][j]=0;
			for(int i=0;i<10;i++)
				cin>>t.m[0][i];
			for(int i=0;i<10;i++)
				t.m[i+1][i]=1;
			n-=9;
			slove();
		}
	}
	return 0;
}