HDU-4565 So Easy!

对于(a+sqrt(b))^n，可以写成X+Y*sqrt(b)的形式，X(1)=a，Y(1)=1

并且可以推出转移公式：X(n)=a*X(n-1)+b*Y(n-1)，Y(n)=X(n-1)+a*Y(n-1)

可以用矩阵快速幂求出X(n)，Y(n)

又因为(a+sqrt(b))^n=Xn+Yn*b，(a-sqrt(b))^n=Xn-Yn*b，且0<(a-sqrt(b))^n<1

又(a+sqrt(b))^n+(a-sqrt(b))^n=2*Xn，所以2*Xn就是(a+sqrt(b))^n向上取整的结果

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
ll MOD;
const int N=2;
struct node
{
	ll a[10][10];
};
node shu,ans,mp;
//shu是输入的矩阵，ans是所求答案
node matrix(node x,node y)
{
    for(int i=1;i<=N;i++)
        for(int j=1;j<=N;j++){
            mp.a[i][j]=0;
            for(int p=1;p<=N;p++)
                mp.a[i][j]=(mp.a[i][j]+x.a[i][p]*y.a[p][j]%MOD+MOD)%MOD;
            //矩阵乘法 
        }
    return mp;
}
void work(ll k)
{//矩阵快速幂 
	for(int i=1;i<=N;i++)
			for(int j=1;j<=N;j++)
				ans.a[i][j]=0;
	for(int i=1;i<=N;i++) ans.a[i][i]=1;
	node t=shu;
    while(k){
        if(k&1) 
            ans=matrix(ans,t);
        k>>=1;
        t=matrix(t,t);
    }
}
int main()
{
	ll a,b,n,m;
	while(~scanf("%lld%lld%lld%lld",&a,&b,&n,&m))
	{
		MOD=m;
		ll ret=0;
		shu.a[1][1]=shu.a[2][2]=a;
		shu.a[2][1]=1;
		shu.a[1][2]=b;
		work(n);
		ret=2*ans.a[1][1];
		ret%=MOD;
		printf("%lld\n",ret);
	}
	return 0;
}