luogu P4139 上帝与集合的正确用法（扩展欧拉定理）

本蒟蒻现在才知带扩展欧拉定理。
对于任意的\(b\geq\varphi(p)\)有
\(a^b\equiv a^{b\ mod\ \varphi(p)+\varphi(p)}(mod\ p)\)
当\(b<\varphi(p)\)有
\(a^b\equiv a^{b\ mod\ \varphi(p)}(mod\ p)\)
\(b\)和\(p\)可以不互质
然后这题就简单了。。。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
using namespace std;
const int N=1e7+100;
#define int long long
int phi[N],prime[N],num,p,t;
bool book[N];
int read(){
	int sum=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){sum=sum*10+ch-'0';ch=getchar();}
	return sum*f;
}
void prework(int n){
	phi[1]=1;
	for(int i=2;i<=n;i++){
		if(book[i]==0){
			phi[i]=i-1;
			prime[++num]=i;
		}
		for(int j=1;i*prime[j]<=n&&j<=num;j++){
			book[i*prime[j]]=1;
			if(i%prime[j]==0){
				phi[i*prime[j]]=phi[i]*prime[j];
				break;
			}
			else phi[i*prime[j]]=phi[i]*(prime[j]-1);
		}
	}
}
int ksm(int x,int b,int mod){
	int tmp=1;
	while(b){
		if(b&1)tmp=tmp*x%mod;
		b>>=1;
		x=x*x%mod; 
	}
	return tmp;
}
int dfs(int x){
	if(x==2)return 0;
	return ksm(2,dfs(phi[x])+phi[x],x);
}
signed main(){
	prework(1e7);
	t=read();
	while(t--){
		p=read();
		printf("%lld\n",dfs(p)); 
	}
	return 0;
}