某 SCOI 模拟赛 T1 迫害 DJ（hakugai）【二次剩余 斐波那契循环节】

题意

$T$ 次询问：已知 $a,b,n,k,mod$，$g_0=a,g_1=b,g_i=g_{i-1}+g_{i-2}$，$f(n,k)=\{\begin{array}{ll}f(g_n,k-1)& (k>0)\\ n& (k=0)\end{array}$，求 $f(n,k)\mathrm{m}\mathrm{o}\mathrm{d}mod$。不保证 $mod$ 为质数。$T\le 10^3$，$0\le a,b<mod\le 10^9$，$k\le 100$。

题解

题目也就是 $g_{g_g\dots }$（套娃）。显然在套娃中直接模 $mod$ 会出问题，考虑套娃时应该模多少，也就是计算 $g$ 在模某数时的循环节。

下文的 $\left(\frac{a}{b}\right)$ 为勒让德符号。当 $a,b$ 都为奇质数时，-1 代表 $a$ 在模 $b$ 下没有二次剩余，1 代表有二次剩余。

$g$ 的特征方程为 $r^2=3r-1$，$r_1=\frac{3+\sqrt{5}}{2},r_2=\frac{3-\sqrt{5}}{2}$。

其通项为 $g_i=C_1{r}_1^i+C_2{r}_2^i$。

设 $cir(n)$ 为模 $n$ 意义下 $g$ 的循环节，我们要计算的就是 $cir$，这样 $g_n\mathrm{m}\mathrm{o}\mathrm{d}x=g_{n\mathrm{m}\mathrm{o}\mathrm{d}cir(x)}\mathrm{m}\mathrm{o}\mathrm{d}x$ 可以大幅缩小数据范围。

小质数

手玩或者打表观察得：$cir(2)=3,cir(3)=8,cir(5)=10$

若 5 在模质数 p 下有二次剩余

判断时：$\left(\frac{5}{p}\right)\left(\frac{p}{5}\right)=-1^{\frac{(p-1)(5-1)}{4}}$，故该情况等价于 $p$ 在模质数 5 下有二次剩余，即 $p\equiv 1\text{ 或 }4(\mathrm{m}\mathrm{o}\mathrm{d}5)$。

$$\begin{array}{rl}{r}_1^p=& \frac{1}{2^p}(3+\sqrt{5}{)}^p\\ =& \frac{1}{2^p}\sum _{i=0}^p{3}^i{\sqrt{5}}^{p-i}\left(\genfrac{}{}{0ex}{}{p}{i}\right)\\ =& \frac{1}{2^p}(3^p+{\sqrt{5}}^p)\\ =& \frac{3+\sqrt{5}}{2}=r_1\end{array}$$

（$\left(\genfrac{}{}{0ex}{}{p}{i}\right)\mathrm{m}\mathrm{o}\mathrm{d}p$ 只在 $i=0$ 或 $i=1$ 时才不为 0）

（最后用的是费马小定理）

同理，$r_2^p=r_2$。

因此 $g_p=g_1$。同理可推得 $g_{p+1}=g_2$，因此 $cir(p)\mid p-1$，计算时取 $cir(p)=p-1$。

若 5 在模质数 p 下没有二次剩余

判断时：$\left(\frac{5}{p}\right)\left(\frac{p}{5}\right)=-1^{\frac{(p-1)(5-1)}{4}}$，故该情况等价于 $p$ 在模质数 5 下没有二次剩余，即 $p\equiv 2\text{ 或 }3(\mathrm{m}\mathrm{o}\mathrm{d}5)$。

$$\begin{array}{rl}{r}_1^p=& \frac{1}{2^p}(3+\sqrt{5}{)}^p\\ =& \frac{1}{2^p}\sum _{i=0}^p{3}^i{\sqrt{5}}^{p-i}\left(\genfrac{}{}{0ex}{}{p}{i}\right)\\ =& \frac{1}{2^p}(3^p+{\sqrt{5}}^p)\\ =& \frac{3+5^{\frac{p-1}{2}}\sqrt{5}}{2}\\ =& \frac{3-\sqrt{5}}{2}=r_2\end{array}$$

$$\begin{array}{rl}{r}_1^{2p+2}=& (r_1^p{)}^2{r}_1^2\\ =& r_2^2{r}_1^2\\ =& (r_1{r}_2{)}^2=1\end{array}$$

同理 $r_2^{2p+2}=1$。因此 $g_0=g_{2p+2}$，同理可推得 $g_{2p+3}=g_1$，因此 $cir(p)\mid 2p+2$，计算时取 $ci{r}_p=2p+2$。

合数

设 $n=\prod _i{p}_i^{a_i}$，则 c i r ( n ) = lcm ⁡ { c i r ( p i ) p i a i − 1 } cir(n)=\operatorname{lcm}\{cir(p_i)p_i^{a_i-1}\} cir(n)=lcm{cir(pi​)piai​−1​}

---

计算完每层的循环节大小就可以矩阵快速幂算了。循环节大小不会大过 long long，$O(\sqrt{n})$ 分解质因数也能跑过去（稍微实现得好一点是跑不满 $O(\sqrt{n})$ 的，因为循环节变大时会引入许多 2 作为约数）。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int getint(){
	int ans=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9'){
		if(c=='-')f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9'){
		ans=ans*10+c-'0';
		c=getchar();
	}
	return ans*f;
}

const int N=2;
ll x[N][N],ans[N][N];
ll tmp[N][N];
ll mod=0;
void qaqx(ll mod){
	for(int i=0;i<2;i++){
		for(int j=0;j<2;j++){
			tmp[i][j]=0;
			for(int k=0;k<2;k++){
				tmp[i][j]=(tmp[i][j]+x[i][k]*1ll*x[k][j]%mod)%mod;
			}
		}
	}
	for(int i=0;i<2;i++)for(int j=0;j<2;j++)x[i][j]=tmp[i][j];
}
void qaqans(ll mod){
	for(int i=0;i<2;i++){
		for(int j=0;j<2;j++){
			tmp[i][j]=0;
			for(int k=0;k<2;k++){
				tmp[i][j]=(tmp[i][j]+ans[i][k]*1ll*x[k][j]%mod)%mod;
			}
		}
	}
	for(int i=0;i<2;i++)for(int j=0;j<2;j++)ans[i][j]=tmp[i][j];
}
void qpow(ll y,ll mod){
	ans[0][0]=ans[1][1]=1;ans[1][0]=ans[0][1]=0;
	x[0][0]=0,x[0][1]=1,x[1][0]=mod-1,x[1][1]=3;
	while(y){
		if(y&1)qaqans(mod);
		qaqx(mod);
		y>>=1;
	}
}
ll gcd(ll x,ll y){
	return x?gcd(y%x,x):y;
}
ll cirp(ll p){
	if(p==2)return 3;
	if(p==5)return 10;
	return (p%5==2||p%5==3)?p*2+2:p-1;
}
ll cir(ll p){
	ll c=1;
	for(ll i=2;i*i<=p&&p!=1;i++){
		if(p%i==0){
			p/=i;
			ll d=1;
			d*=cirp(i);
			while(p%i==0)p/=i,d*=i;
			c=c/gcd(c,d)*d;
		}
	}
	if(p!=1)c=c/gcd(c,cirp(p))*cirp(p);
	//cerr<<"> "<<p<<" "<<c<<endl;
	return c;
}
ll g(ll a,ll b,ll n,ll p){
	qpow(n,p);
	return (a*1ll*ans[0][0]%p+b*1ll*ans[0][1]%p)%p;
}

ll q[233];
int main(){
	int T=getint();
	while(T--){
		ll a=getint(),b=getint(),n=getint(),k=getint(),p=getint();
		q[0]=p;
		for(int i=1;i<=k;i++)q[i]=cir(q[i-1]);n%=q[k];
		for(int i=k-1;i>=0;--i)n=g(a,b,n,q[i]);
		printf("%d\n",int(n));
	}
	return 0;
}