2023牛客暑期多校训练营9-B Semi-Puzzle: Brain Storm

2023牛客暑期多校训练营9-B Semi-Puzzle: Brain Storm

https://ac.nowcoder.com/acm/contest/57363/B

题意

解题思路

欧拉定理
$$a^b\equiv \{\begin{array}{llc}& a^{b\%\phi (p)}& gcd(a,p)=1\\ & a^b& gcd(a,p)\ne 1,b<\phi (p)\\ & a^{b\%\phi (p)+\phi (p)}& gcd(a,p)\ne 1,b\ge \phi (p)\end{array}$$
$$\begin{array}{c}{a}^u\equiv u(\mathrm{mod}p)\\ 设d=u\%\phi (p)+\phi (p)\\ u=d+k\phi (p)\\ a^{d+k\phi (p)}\equiv d+k\phi (p)(\mathrm{mod}p)\\ a^d-d\equiv k\phi (p)(\mathrm{mod}p)\end{array}$$
将取余打开，可得：
$$\begin{array}{c}\phi (p)x+py=a^d-d\end{array}$$
显然可以用扩展欧几里得求解当$\phi (p)x+py=\gcd (p,\varphi (p))$的解，为保证$d$有解，故$\gcd (p,\phi (p))\mid a^d-d$，设$a^d-d=h\gcd (\phi (p),p)$，故$a^d=h\gcd (\phi (p),p)+d$，可以发现$a^d\equiv d(\mathrm{mod}\gcd (\phi (p),p))$，可以发现形式上与$a^u\equiv u(\mathrm{mod}p)$，显然当$p=1$时，$u=0$，有了边界条件，可以递归求出$u$。$u=d+k\phi (p)$，$k$即为$\phi (p)x+py=a^d-d$中$x$的解，当求出$\phi (p)x_0+p{y}_0=\gcd (p,\varphi (p))$:
$$\begin{array}{cc}x=& x_0\times (\frac{a^d-d}{\gcd (\phi (p),p)}\%p)\\ =& x_0\times (\frac{(a^d-d)\%p}{\gcd (\phi (p),p)})\end{array}$$

代码

#include
#define ll long long
#define pii pair
using namespace std;
ll T,a,m;
ll phi(ll n){
	ll sum=n;
	for(ll i=2;i*i<=n;i++){
		if(n%i==0){
			while(n%i==0)n/=i;
			sum/=i,sum*=i-1;
		}
	}
	
	if(n>1)return sum/n*(n-1);
	return sum;
}
ll exgcd(ll a,ll b,ll &x,ll &y){
	if(!b){
		x=1,y=0;
		return a;
	}
	ll v=exgcd(b,a%b,y,x);
	y-=a/b*x;
	return v;
}
ll power(ll x,ll p,ll mod){
	ll res=1;
	while(p){
		if(p&1)res*=x,res%=mod;
		x*=x,x%=mod;
		p>>=1;
	}
	return res;
}
ll dfs(ll a,ll p){
	if(p==1)return 0;
	ll ph=phi(p);
	ll x,y;
	ll v=exgcd(ph,p,x,y);
	x=(x%p+p)%p;
	ll d=dfs(a,v)+ph;
	return d+(x*((power(a%p,d,p)-d%p+p)%p/v)%p)*ph;
}
void solve(){
	cin>>a>>m;
	ll x=dfs(a,m);
	cout<>T;
	while(T--)solve();
}