bzoj 3884 上帝与集合的正确用法 扩展欧拉定理

Description

根据一些书上的记载，上帝的一次失败的创世经历是这样的：
第一天， 上帝创造了一个世界的基本元素，称做“元”。
第二天， 上帝创造了一个新的元素，称作“α”。“α”被定义为“元”构成的集合。容易发现，一共有两种不同的“α”。
第三天， 上帝又创造了一个新的元素，称作“β”。“β”被定义为“α”构成的集合。容易发现，一共有四种不同的“β”。
第四天， 上帝创造了新的元素“γ”，“γ”被定义为“β”的集合。显然，一共会有16种不同的“γ”。
如果按照这样下去，上帝创造的第四种元素将会有65536种，第五种元素将会有2^65536种。这将会是一个天文数字。
然而，上帝并没有预料到元素种类数的增长是如此的迅速。他想要让世界的元素丰富起来，因此，日复一日，年复一年，他重复地创造着新的元素……
然而不久，当上帝创造出最后一种元素“θ”时，他发现这世界的元素实在是太多了，以致于世界的容量不足，无法承受。因此在这一天，上帝毁灭了世界。
至今，上帝仍记得那次失败的创世经历，现在他想问问你，他最后一次创造的元素“θ”一共有多少种？
上帝觉得这个数字可能过于巨大而无法表示出来，因此你只需要回答这个数对p取模后的值即可。
你可以认为上帝从“α”到“θ”一共创造了10^9次元素，或10^18次，或者干脆∞次。
一句话题意：

Input

接下来T行，每行一个正整数p，代表你需要取模的值
Output

T行，每行一个正整数，为答案对p取模后的值
Sample Input

3

2

3

6
Sample Output

0

1

4
HINT

对于100%的数据，T<=1000,p<=10^7

传送门
扩展欧拉定理裸题。
$当y>=\phi{P}时,x^y=x^{y\%\phi{P}+\phi{P}}$
$当y<\phi{P}时,x^y=x^{y\%\phi{P}}$
一个数求$\phi$ 很快就会到1了，
不超过log次，
所以取出$\phi{P}$ 的序列从上往下判断就好了。

需要快速幂，里面加入了判断是否溢出下一层的欧拉函数值。

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int
    MAX=1e7;
int tl,pcnt,a[30],prime[500000];
bool notprime[MAX];
void Get_Prime(){
    notprime[1]=1,pcnt=0;
    for (int i=2;i<MAX;i++){
        if (!notprime[i]) prime[++pcnt]=i;
        for (int j=1;j<=pcnt;j++){
            if (prime[j]*i>=MAX) break;
            notprime[prime[j]*i]=1;
            if (!(i%prime[j])) break;
        }
    }
}
int phi(int x){
    int t=x,j=1;
    while (j<=pcnt && x!=1){
        if (!(x%prime[j])){
            t=t/prime[j]*(prime[j]-1);
            while (!(x%prime[j])) x/=prime[j];
        }
        j++;
    }
    if (x!=1) t=t/x*(x-1);
    return t;
}
int ksm(int y,int z,bool &f){
    int a=1,x=2;f=0;
    while (y){
        if (y&1){
            if ((ll)a*x>=z) f=1;
            a=(ll)a*x%z;
        }
        y>>=1;
        if ((ll)x*x>=z) f=1;
        x=(ll)x*x%z;
    }
    return a;
}
int calc(){
    int x=1;bool f;
    for (int i=tl-1;~i;i--){
        x=ksm(x,a[i],f);
        if (f) x+=a[i];
    }
    return x%a[0];
}
int main(){
    Get_Prime();
    int cas,P;scanf("%d",&cas);
    bool x;
    while (cas--){
        scanf("%d",&P);
        if (P==1){puts("0");continue;}
        a[tl=0]=P;
        while (P!=1){
            a[++tl]=phi(P);
            P=phi(P);
        }
        printf("%d\n",calc());
    }
    return 0;
}