HDU4565(数学)

构造An=(a+sqrt(b)^n+(a-sqrt(b))^n，An为整数。

由于b的范围，求An即可。

An*((a+sqrt(b))+(a-sqrt(b)))=...=2*a*An=...=A(n+1)+(a*a-b)*A(n-1),

得到递推关系A(n+1)=2*a*An+(b-a*a)*A(n-1)。

构造矩阵temp[2][2]={2*a,-a*a+b,1,0}.

使用矩阵快速幂求解即可

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define N 2
#define ll long long
ll a,b,n,m;
void mul(ll a[N][N],ll b[N][N])//矩阵乘法  
{  
    ll ret[N][N]={0};  
    int i,j,k;  
    for(i=0;i<N;i++)  
    {  
        for(j=0;j<N;j++)  
        {  
            for(k=0;k<N;k++)  
            {  
                ret[i][j]+=a[i][k]*b[k][j];  
                ret[i][j]%=m;
            }  
        }  
    }  
    for(i=0;i<N;i++)  
    {  
        for(j=0;j<N;j++)a[i][j]=ret[i][j];  
    }  
}  
int main()
{
	while(scanf("%I64d%I64d%64d%64d",&a,&b,&n,&m)!=EOF)
	{
		ll ans[N][N]={2*a,-2*a*a+2*b,0,0};
		ll temp[N][N]={2*a,-a*a+b,1,0};
		n--;
		while(n)
		{
			if(n&1)mul(ans,temp);
			mul(temp,temp);
			n>>=1;
		}
		printf("%I64d\n",(ans[0][0]+m)%m);
	}
	return 0;
}