题解:
因为b有一个范围,即a-1<b<a,那么就有0<a-sqrt(b)<1,而(a+sqrt(b))^n+(a-sqrt(b))^n为整数,于是可以知道(a+sqrt(b))^n上取整等于(a+sqrt(b))^n+(a-sqrt(b))^n,记其为Cn,
Cn*((a+sqrt(b))+(a-sqrt(b))),可得递推公式Cn+1=2a*Cn-(a^2-b)*Cn-1,而Pn+1=aPn+bPn-1,这种形式可以利用矩阵快速幂求得,即构造矩阵
答案即可求得
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#define ll long long
using namespace std;
typedef struct
{
ll m[2][2];
}node;
node p;
node mul(node a,node b,ll mod)
{
node ans;
for(int i=0;i<2;i++)
for(int j=0;j<2;j++)
ans.m[i][j]=0;
for(int i=0;i<2;i++)
for(int j=0;j<2;j++)
for(int k=0;k<2;k++)
ans.m[i][j]=(a.m[i][k]*b.m[k][j]%mod+ans.m[i][j])%mod;
return ans;
}
node quickPow(node a,ll k,ll mod)
{
node ans;
for(int i=0;i<2;i++)
for(int j=0;j<2;j++)
ans.m[i][j]=(i==j);
while(k)
{
if(k&1)
ans=mul(ans,a,mod);
a=mul(a,a,mod);
k=k>>1;
}
return ans;
}
int main()
{
ll a,b,n,m;
while(scanf("%I64d%I64d%I64d%I64d",&a,&b,&n,&m)!=EOF)
{
node base;
base.m[0][0]=2*a;
base.m[0][1]=b-a*a;
base.m[1][0]=1;
base.m[1][1]=0;
node ans=quickPow(base,n-1,m);
ll t=(ans.m[0][0]*2*a%m+ans.m[0][1]*2%m)%m;
printf("%I64d\n",(t+m)%m);
}
return 0;
}