hdu 4990 Reading comprehension (矩阵快速幂)

题目链接 

根据题目中给的程序，跑出来前几个数，推导一下

如果 i 为奇数 
ans = (ans * 2 + 1) % m 
反之 
ans = ans * 2 % m

矩阵快速幂加速即可

#include<bits/stdc++.h>
using namespace std;
const int maxn=3000+5;

typedef long long ll;
ll n,m;
struct matrix
{
	ll 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])%m;
		return res;
	}
}ans,base;

void init()
{
	base.a[0][0]=4;base.a[0][1]=0;
	base.a[1][0]=2;base.a[1][1]=1;
	ans.a[0][1]=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[])
{
	while(scanf("%lld%lld",&n,&m)!=EOF)
	{
		memset(ans.a,0,sizeof(ans.a));
		memset(base.a,0,sizeof(base.a));
		//memset(ans.a,0,sizeof(ans.a));
		if(n==1){printf("%lld\n",1%m);continue;}
		if(n==2){printf("%lld\n",2%m);continue;}
		init();qpow(n>>1);
		if(n&1)printf("%lld\n",(ans.a[0][0]<<1|1)%m);
		else printf("%lld\n",ans.a[0][0]%m);
	}
	return 0;
}