Pow

题目描述

定义a^b为a的b次方，并且^是满足右结合的，即a^b^c^d=a^(b^(c^d))。例如，2^3^2=2^(3^2)=2^9=512。
现在给定n个数a1,a2,…,an 求a1^a2^…^an对p取模的值。

事后补充

Drin_E大爷发现下面的方法有错因为$(d,p')$不一定为1。
唉不管了反正n最大20，你们想保证正确性就用相同方法递归求解呗（逃，因为$d^{x-1}\% p'$也是形如$a^x\% p$的。

一个东西

我们只需要考虑如何计算$a^x\% p$
那么就可以解决这道题。
设$d=(a,p),a'=a/d,p'=p/d,ans=a^x\% p$
也就是$a^x\equiv ans(mod \ p)$
$d^x*a'^x\equiv ans(mod \ d*p')$
两端同时除以d
$d^{x-1}*a'^x\equiv \frac{ans}{d}(mod \ p')$
那么此时$(a',p')=1$，说明$a'^{\phi(p')}\equiv1(mod \ p')$
那么我们可以计算$x\%\phi(p')$
需要注意在模意义下$d^{-1}\equiv d^{\phi(p')-1}(mod \ p')$

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
typedef long long ll;
const int maxd=10000000+10;
int pri[maxd],phi[maxd],b[30];
bool bz[maxd];
int i,j,k,l,t,n,m,p,top,ca;
int quicksortmi(int x,int y,int p){
    if (!y) return 1;
    int t=quicksortmi(x,y/2,p);
    t=(ll)t*t%p;
    if (y%2) t=(ll)t*(x%p)%p;
    return t;
}
int gcd(int a,int b){
    if (!b) return a;else return gcd(b,a%b);
}
int solve(int i,int p){
    if (i==n+1) return 1;
    int a=b[i],d=gcd(a,p);
    a/=d;p/=d;
    int x=solve(i+1,phi[p]);
    int t=quicksortmi(a,x,p);
    if (x) t=(ll)t*quicksortmi(d,x-1,p)%p;else t=(ll)t*quicksortmi(d,phi[p]-1,p)%p;
    return t*d;
}
int main(){
    phi[1]=1;
    fo(i,2,maxd-10){
        if (!bz[i]){
            pri[++top]=i;
            phi[i]=i-1;
        }
        fo(j,1,top){
            if ((ll)i*pri[j]>maxd-10) break;
            bz[i*pri[j]]=1;
            if (i%pri[j]==0){
                phi[i*pri[j]]=phi[i]*pri[j];
                break;
            }
            phi[i*pri[j]]=phi[i]*(pri[j]-1);
        }
    }
    scanf("%d",&ca);
    while (ca--){
        scanf("%d%d",&n,&p);
        fo(i,1,n) scanf("%d",&b[i]);
        printf("%d\n",solve(1,p));
    }
}