2019ICPC南京网络赛B super_log——扩展欧拉定理

题目

设函数

$$log_a*(x) = \begin{cases}
-1, & \text{ if } x < 1 \\
1+log_a*(log_ax) & \text{ if } x \geq 1
\end{cases}$$

求最小的正整数 $x$，使得 $log_a*(x) \geq b$

分析

通过将递归式展开，展开 $b$ 次等于1，所以 $x$ 为 $a^{a^{a^{...}}}$（共 $b$ 次）

由欧拉降幂公式

$$a^b= \begin{cases} a^{b \% \varphi(p)} &  gcd(a,p)=1 \\  a^b &  gcd(a,p)\neq 1,b < \varphi (p) \\  a^{b\% \varphi (p) + \varphi (p)} & gcd(a,p)\neq 1,b \geq \varphi (p) \end{cases}$$

用这个公式，需要讨论 $b$ 与 $\varphi(p)$ 的大小关系，很麻烦。

看网上的博客，有一种精妙的方法，只需重写 Mod 函数，就能当作 $a$ 与 $p$ 互素处理。证明见博客。

要点：

• 快速幂和cal函数都要换成Mod；
• 最终答案需要模p

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
ll a, b, p;

ll Mod(ll x, ll mod)
{
    return x < mod ? x : x % mod + mod;
}

ll euler_phi(ll n)
{
    ll m = (ll)sqrt(n + 0.5);
    ll ans = n;
    for (ll i = ; i <= m; i++)
    {
        if (n % i == )
        {
            ans = ans / i * (i - );
            while (n % i == )  n /= i;        //除尽
        }
    }
    if (n > )  ans = ans / n * (n - );    //剩下的不为1,也是素数
    return ans;
}

ll qpow(ll a, ll b, ll p)
{
    ll ret = ;
    while(b)
    {
        if(b&) ret = Mod(ret * a, p);
        a = Mod(a * a ,p);
        b >>= ;
    }
    return ret;
}

ll cal(ll a, ll b, ll p)   //a^a^a..^a共b次
{
   // printf("%lld %lld\n", t, p);
    //if(t == 1)  return Mod(a, p);
    if(b == )  return Mod(, p);
    if(p == )  return Mod(a, p);
    ll phip = euler_phi(p);
    return qpow(a, cal(a, b-, phip), p);  //第一类和第三类
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%lld%lld%lld", &a, &b, &p);
        printf("%lld\n", cal(a, b, p) % p);  //这个取模不能少
    }
    return ;
}

之前写了一种需要讨论 $b$  与 $\varphi(p)$ 大小的，

因为模数 $p \leq 1e6$，可以找找 $b$ 小于 $\varphi(p)$ 的情况

打表发现，

当次数大于3时，只有 $a=2$ 可能小于 $\varphi(p)$，特判一下就好了。

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;
ll a, b, p;
int table[] = {, , , ,};

ll qpow(ll a, ll b, ll p)
{
    a = a%p;
    ll ret = %p;
    while(b)
    {
        if(b&) ret = ret*a%p;
        a = a*a%p;
        b >>= ;
    }
    return ret%p;
}

ll euler_phi(ll n)
{
    ll m = (ll)sqrt(n + 0.5);
    ll ans = n;
    for (ll i = ; i <= m; i++)
    {
        if (n % i == )
        {
            ans = ans / i * (i - );
            while (n % i == )  n /= i;        //除尽
        }
    }
    if (n > )  ans = ans / n * (n - );    //剩下的不为1,也是素数
    return ans;
}

ll f(ll p, ll t)
{
    //printf("%lld %lld\n", p, t);
    if(t == )  return a%p;
    if(t == )  return qpow(a, a, p);
    if(t == )  return %p;
    if(p == )  return ;
    ll phip = euler_phi(p);
    if(p % a == )  //t >= 3，若出现第二种情况，a只能为2
    {
        if(a ==  && t <=  && table[t-] < phip)
            //return qpow(a, f(p, t-1), p);
            return table[t]%p;
    }
    return qpow(a, f(phip, t-)+phip, p);  //第一类和第三类
}

int main()
{
    int T;
    scanf("%d", &T);
    while(T--)
    {
        scanf("%lld%lld%lld", &a, &b, &p);
        printf("%lld\n", f(p, b));
    }
    return ;
}

参考链接：https://blog.csdn.net/qq_35914587/article/details/79883547