UVA - 10689 Yet another Number Sequence (矩阵快速幂)

题目链接

基本是裸题，取模要处理一下

用一个mod数组记录值 int md[] = {0, 10, 100, 1000, 10000};

#include<bits/stdc++.h>
using namespace std;
const int maxn=3000+5;
//const int mod=1e9+7;
typedef long long ll;
int a,b,n,m;
int md[] = {0, 10, 100, 1000, 10000};
int mod;
struct matrix
{
	int a[2][2];
	matrix (){memset(a,0,sizeof(a));}
	matrix operator *(const matrix &b)const
	{
		matrix res;
		for(int i=0;i<2;i++)
			for(int j=0;j<2;j++)
				for(int k=0;k<2;k++)
					res.a[i][j]=(res.a[i][j]+a[i][k]*b.a[k][j])%mod;
		return res;
	}
}ans,base;

void init()
{
	base.a[0][0]=base.a[0][1]=base.a[1][0]=1;
	ans.a[1][1]=ans.a[0][0]=1;
}
void qpow(int b)
{
	while(b)
	{
		if(b&1)ans=ans*base;
		base=base*base;
		b>>=1;
	}
}
int main(int argc, char const *argv[])
{
	int t;scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d%d%d",&a,&b,&n,&m);
		memset(base.a,0,sizeof(base.a));
		memset(ans.a,0,sizeof(ans.a));
		mod=md[m];
		if(n==0) {printf("%d",a%mod);continue;}
		if(n==1){printf("%d",b%mod);continue;}
		init();
		qpow(n-1);
		printf("%d\n",(b*ans.a[0][0]+a*ans.a[1][0])%mod);
	}
	return 0;
}