类斐波拉契数列的循环节
设转移矩阵
A=(a1b0)
A 的特征值为 |λE−A| = λ2−aλ−b = 0
所以 A 的特征值为
λ1,2=a±a2+4b−−−−−−√2
设 c=a2+4b
(1) c 是模 p 的二次剩余,n = p - 1
(2) c 是模 p 的非二次剩余,n = (p + 1)(p - 1)
所以枚举 n 的因子就可以找到最小循环节了
题目大意:求G(a,b,n,p)=(ap−12+1)(bp−12+1)[(a√+b√)2Fn+(a√−b√)2Fn](modp)
原式 = (ap−12+1)(bp−12+1)[(a+b+ab−−√)Fn+(a+b−ab−−√)Fn](modp)
前一部分可以快速幂计算,若不为零则可以知道是二次剩余,p - 1为循环节,所以 Fn 可以对 p−1 取模
后面的式子可以化简得到递推式
Gn=2(a+b)Gn−1−(a−b)2Gn−2
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<string>
#include<cstring>
#include<ctime>
#include<cmath>
#define LL long long
using namespace std;
LL T,a,b,n,p;
LL ret[2][2],A[2][2],t[2][2];
LL ksm(LL a,LL b)
{
LL ret = 1;
for (;b;b >>= 1,a = a * a % p)
if (b & 1) ret = ret * a % p;
return ret;
}
LL mul(LL a[][2],LL b[][2],LL p)
{
for (int i = 0;i < 2;i ++)
for (int j = 0;j < 2;j ++)
{
t[i][j] = 0;
for (int k = 0;k < 2;k ++)
(t[i][j] += a[i][k] * b[k][j]) %= p;
}
for (int i = 0;i < 2;i ++)
for (int j = 0;j < 2;j ++)
a[i][j] = t[i][j];
}
LL fib(LL n)
{
n --;
ret[0][0] = ret[1][1] = A[1][0] = A[0][0] = A[0][1] = 1;
ret[0][1] = ret[1][0] = A[1][1] = 0;
for (;n;n >>= 1,mul(A,A,p-1))
if (n & 1) mul(ret,A,p-1);
return (ret[0][0] + ret[0][1]) % (p - 1);
}
LL mat_ksm(LL n)
{
if (!n) n = p - 1;n --;
ret[0][0] = ret[1][1] = A[1][0] = 1;
ret[0][1] = ret[1][0] = A[1][1] = 0;
A[0][0] = (a+b)*2%p,A[0][1] = -(a - b) * (a - b) % p;
for (;n;n >>= 1,mul(A,A,p))
if (n & 1) mul(ret,A,p);
return ((a+b)*2*ret[0][0]+ret[0][1]*2)%p;
}
int main()
{
cin >> T;
while (T --)
{
cin >> a >> b >> n >> p;
LL ans = (ksm(a,p >> 1) + 1) * (ksm(b,p >> 1) + 1) % p;
if (ans == 0) {puts("0");continue;}
if (n < 2) {printf("%d\n",(a+b)*2%p*ans%p);continue;}
(ans *= mat_ksm(fib(n))) %= p;
if (ans < 0) ans += p;
cout << ans << endl;
}
return 0;
}