src/alfred/math/speed-of-light-power.hpp

• View this file on GitHub
• Last update: 2025-03-29 12:01:17+08:00
• Include: #include "src/alfred/math/speed-of-light-power.hpp"

Depends on

• src/alfred/math/utils.hpp

Code

#ifndef AFMT_SPEED_OF_LIGHT_POWER
#define AFMT_SPEED_OF_LIGHT_POWER

#include "utils.hpp"
#include <cmath>
#include <vector>

template <int base, int mod>
struct SOLPower { // Speed Of Light Power. O(sqrt(n)) ~ O(1)
    // p1 stores base^0  ~ base^sq
    // ps stores base^sq ~ base^(sq^2)
    std::vector<int> p1, ps;
    static const bool P = is_prime(mod);
    static const int sq = std::sqrt(mod);
    SOLPower(void) {
        p1.push_back(1), ps.push_back(1);
        for (int i = 1; i <= sq; i++) {
            p1.push_back(1ll * p1.back() * base % mod);
        }
        ps.push_back(p1.back());
        for (int i = 2 * sq; i <= mod; i += sq) {
            ps.push_back(1ll * ps.back() * ps[1] % mod);
        }
    }
    inline int power(long long index) {
        if (P && index >= mod) index %= mod;
        return 1ll * ps[index / sq] * p1[index % sq] % mod;
    }
};

#endif // AFMT_SPEED_OF_LIGHT_POWER

#line 1 "src/alfred/math/speed-of-light-power.hpp"



#line 1 "src/alfred/math/utils.hpp"



// O(sqrt(x)) to judge if x is a prime.
// It will be re-written in Miller-rabin sometime.
constexpr bool is_prime(long long x) {
    if (x < 2) return false;
    for (long long i = 2; i * i <= x; i++) {
        if (x % i == 0) return false;
    }
    return true;
}

template <class T>
inline T gcd(T a, T b) {
    return b == 0 ? a : gcd(b, a % b);
}

template <class T>
inline T lcm(T a, T b) {
    return a / gcd(a, b) * b;
}


#line 5 "src/alfred/math/speed-of-light-power.hpp"
#include <cmath>
#include <vector>

template <int base, int mod>
struct SOLPower { // Speed Of Light Power. O(sqrt(n)) ~ O(1)
    // p1 stores base^0  ~ base^sq
    // ps stores base^sq ~ base^(sq^2)
    std::vector<int> p1, ps;
    static const bool P = is_prime(mod);
    static const int sq = std::sqrt(mod);
    SOLPower(void) {
        p1.push_back(1), ps.push_back(1);
        for (int i = 1; i <= sq; i++) {
            p1.push_back(1ll * p1.back() * base % mod);
        }
        ps.push_back(p1.back());
        for (int i = 2 * sq; i <= mod; i += sq) {
            ps.push_back(1ll * ps.back() * ps[1] % mod);
        }
    }
    inline int power(long long index) {
        if (P && index >= mod) index %= mod;
        return 1ll * ps[index / sq] * p1[index % sq] % mod;
    }
};

Back to top page