MtSaka's Library

This documentation is automatically generated by online-judge-tools/verification-helper

View the Project on GitHub MtSaka/library

:heavy_check_mark: test/yosupo/number_theory/discrete_logarithm_mod.test.cpp

Depends on

Required by

Code

// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/discrete_logarithm_mod
#include "../../../math/modular/mod-log.hpp"
#include "../../../template/template.hpp"
int main() {
    int t;
    sc >> t;
    rep(t) {
        ll x, y, m;
        sc >> x >> y >> m;
        print(discrete_logarithm(x, y, m));
    }
}
#line 1 "test/yosupo/number_theory/discrete_logarithm_mod.test.cpp"
// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/discrete_logarithm_mod
#line 2 "template/template.hpp"
#include <bits/stdc++.h>

#line 3 "template/alias.hpp"

using ll = long long;
using ull = unsigned long long;
using ld = long double;
using i128 = __int128_t;
using u128 = __uint128_t;
using pi = std::pair<int, int>;
using pl = std::pair<ll, ll>;
using vi = std::vector<int>;
using vl = std::vector<ll>;
using vs = std::vector<std::string>;
using vc = std::vector<char>;
using vvl = std::vector<vl>;
using vd = std::vector<double>;
using vp = std::vector<pl>;
using vb = std::vector<bool>;
template <typename T>
struct infinity {
    static constexpr T max = std::numeric_limits<T>::max();
    static constexpr T min = std::numeric_limits<T>::min();
    static constexpr T value = std::numeric_limits<T>::max() / 2;
    static constexpr T mvalue = std::numeric_limits<T>::min() / 2;
};
template <typename T>
constexpr T INF = infinity<T>::value;
constexpr ll inf = INF<ll>;
constexpr ld EPS = 1e-8;
constexpr ld PI = 3.1415926535897932384626;
constexpr int dx[8] = {-1, 0, 1, 0, 1, -1, -1, 1};
constexpr int dy[8] = {0, 1, 0, -1, 1, 1, -1, -1};
#line 3 "template/macro.hpp"

#ifndef __COUNTER__
#define __COUNTER__ __LINE__
#endif

#define SELECT4(a, b, c, d, e, ...) e
#define SELECT3(a, b, c, d, ...) d
#define REP_1(a, c) for (ll REP_##c = 0; REP_##c < (ll)(a); ++REP_##c)
#define REP1(a) REP_1(a, __COUNTER__)
#define REP2(i, a) for (ll i = 0; i < (ll)(a); ++i)
#define REP3(i, a, b) for (ll i = (ll)(a); i < (ll)(b); ++i)
#define REP4(i, a, b, c) for (ll i = (ll)(a); i < (ll)(b); i += (ll)(c))
#define rep(...) SELECT4(__VA_ARGS__, REP4, REP3, REP2, REP1)(__VA_ARGS__)
#define RREP_1(a, c) for (ll RREP_##c = (ll)(a) - 1; RREP_##c >= 0; --RREP_##c)
#define RREP1(a) RREP_1(a, __COUNTER__)
#define RREP2(i, a) for (ll i = (ll)(a) - 1; i >= 0; --i)
#define RREP3(i, a, b) for (ll i = (ll)(b) - 1; i >= (ll)(a); --i)
#define rrep(...) SELECT3(__VA_ARGS__, RREP3, RREP2, RREP1)(__VA_ARGS__)
#define all(v) std::begin(v), std::end(v)
#define rall(v) std::rbegin(v), std::rend(v)
#define INT(...)     \
    int __VA_ARGS__; \
    scan(__VA_ARGS__)
#define LL(...)     \
    ll __VA_ARGS__; \
    scan(__VA_ARGS__)
#define STR(...)        \
    string __VA_ARGS__; \
    scan(__VA_ARGS__)
#define CHR(...)      \
    char __VA_ARGS__; \
    scan(__VA_ARGS__)
#define DBL(...)        \
    double __VA_ARGS__; \
    scan(__VA_ARGS__)
#define LD(...)     \
    ld __VA_ARGS__; \
    scan(__VA_ARGS__)
#define pb push_back
#define eb emplace_back
#line 3 "template/type-traits.hpp"

#line 5 "template/type-traits.hpp"

template <typename T, typename... Args>
struct function_traits_impl {
    using return_type = T;
    static constexpr std::size_t arg_size = sizeof...(Args);
    template <std::size_t idx>
    using argument_type = typename std::tuple_element<idx, std::tuple<Args...>>::type;
    using argument_types = std::tuple<Args...>;
};

template <typename>
struct function_traits_helper;
template <typename T, typename Tp, typename... Args>
struct function_traits_helper<T (Tp::*)(Args...)> : function_traits_impl<T, Args...> {};
template <typename T, typename Tp, typename... Args>
struct function_traits_helper<T (Tp::*)(Args...) const> : function_traits_impl<T, Args...> {};
template <typename T, typename Tp, typename... Args>
struct function_traits_helper<T (Tp::*)(Args...)&> : function_traits_impl<T, Args...> {};
template <typename T, typename Tp, typename... Args>
struct function_traits_helper<T (Tp::*)(Args...) const&> : function_traits_impl<T, Args...> {};

template <typename F>
using function_traits = function_traits_helper<decltype(&std::remove_reference<F>::type::operator())>;
template <typename F>
using function_return_type = typename function_traits<F>::return_type;
template <typename F, std::size_t idx>
using function_argument_type = typename function_traits<F>::template argument_type<idx>;
template <typename F>
using function_argument_types = typename function_traits<F>::argument_types;
template <class T>
using is_signed_int = std::integral_constant<bool, (std::is_integral<T>::value && std::is_signed<T>::value) || std::is_same<T, __int128_t>::value>;
template <class T>
using is_unsigned_int = std::integral_constant<bool, (std::is_integral<T>::value && std::is_unsigned<T>::value) || std::is_same<T, __uint128_t>::value>;
template <class T>
using is_int = std::integral_constant<bool, is_signed_int<T>::value || is_unsigned_int<T>::value>;
template <typename T, typename = void>
struct is_range : std::false_type {};
template <typename T>
struct is_range<
    T,
    decltype(all(std::declval<typename std::add_lvalue_reference<T>::type>()), (void)0)> : std::true_type {};
template <std::size_t size>
struct int_least {
    static_assert(size <= 128, "size must be less than or equal to 128");
    using type = typename std::conditional<
        size <= 8, std::int_least8_t,
        typename std::conditional<
            size <= 16, std::int_least16_t,
            typename std::conditional<
                size <= 32, std::int_least32_t,
                typename std::conditional<size <= 64, std::int_least64_t, __int128_t>::type>::type>::type>::type;
};
template <std::size_t size>
using int_least_t = typename int_least<size>::type;
template <std::size_t size>
struct uint_least {
    static_assert(size <= 128, "size must be less than or equal to 128");
    using type = typename std::conditional<
        size <= 8, std::uint_least8_t,
        typename std::conditional<
            size <= 16, std::uint_least16_t,
            typename std::conditional<
                size <= 32, std::uint_least32_t,
                typename std::conditional<size <= 64, std::uint_least64_t, __uint128_t>::type>::type>::type>::type;
};
template <std::size_t size>
using uint_least_t = typename uint_least<size>::type;
template <typename T>
using double_size_int = int_least<std::numeric_limits<T>::digits * 2 + 1>;
template <typename T>
using double_size_int_t = typename double_size_int<T>::type;
template <typename T>
using double_size_uint = uint_least<std::numeric_limits<T>::digits * 2>;
template <typename T>
using double_size_uint_t = typename double_size_uint<T>::type;
template <typename T>
using double_size = typename std::conditional<std::is_signed<T>::value, double_size_int<T>, double_size_uint<T>>::type;
template <typename T>
using double_size_t = typename double_size<T>::type;
#line 2 "template/in.hpp"
#include <unistd.h>
#line 5 "template/in.hpp"
namespace fastio {
template <std::size_t BUFF_SIZE = 1 << 17, int decimal_precision = 16>
struct Scanner {
   private:
    template <typename, typename = void>
    struct has_scan : std::false_type {};
    template <class T>
    struct has_scan<T, decltype(std::declval<T>().scan(std::declval<Scanner&>()), (void)0)> : std::true_type {};
    int fd;
    char buffer[BUFF_SIZE + 1];
    int idx, sz;
    bool state;
    inline void load() {
        int len = sz - idx;
        if (idx < len) return;
        std::memcpy(buffer, buffer + idx, len);
        sz = len + read(fd, buffer + len, BUFF_SIZE - len);
        idx = 0;
        buffer[sz] = 0;
    }
    inline char cur() {
        if (idx == sz) load();
        if (idx == sz) {
            state = false;
            return '\0';
        }
        return buffer[idx];
    }
    inline void next() {
        if (idx == sz) load();
        if (idx == sz) return;
        idx++;
    }

   public:
    Scanner() : Scanner(0) {}
    explicit Scanner(int fd) : fd(fd), idx(0), sz(0), state(true) {}
    explicit Scanner(FILE* file) : fd(fileno(file)), idx(0), sz(0), state(true) {}

    inline char scan_char() {
        if (idx == sz) load();
        return (idx == sz ? '\0' : buffer[idx++]);
    }
    Scanner ignore(int n = 1) {
        if (idx + n > sz) load();
        idx += n;
        return (*this);
    }
    inline void skip_space() {
       
        if (idx == sz) load();
        while (('\t' <= cur() && cur() <= '\r') || cur() == ' ') {
            if (++idx == sz) load();
        }
    }
    void scan(char& a) {
        skip_space();
        a = scan_char();
    }
    void scan(std::string& a) {
        skip_space();
        a.clear();
        while (cur() != '\0' && (buffer[idx] < '\t' || '\r' < buffer[idx]) && buffer[idx] != ' ') {
            a += scan_char();
        }
    }
    template <std::size_t len>
    void scan(std::bitset<len>& a) {
        skip_space();
        if (idx + len > sz) load();
        rrep(i, len) a[i] = (buffer[idx++] != '0');
    }
    template <typename T, typename std::enable_if<is_int<T>::value && !has_scan<T>::value>::type* = nullptr>
    void scan(T& a) {
        skip_space();
        bool neg = false;
        if constexpr (std::is_signed<T>::value || std::is_same_v<T, __int128_t>) {
            if (cur() == '-') {
                neg = true;
                next();
            }
        }
        if (idx + 40 > sz && (idx == sz || ('0' <= buffer[sz - 1] && buffer[sz - 1] <= '9'))) load();
        a = 0;
        while ('0' <= buffer[idx] && buffer[idx] <= '9') {
            a = a * 10 + (buffer[idx++] & 15);
        }
        if constexpr (std::is_signed<T>::value || std::is_same<T, __int128_t>::value) {
            if (neg) a = -a;
        }
    }
    template <typename T, typename std::enable_if<std::is_floating_point<T>::value && !has_scan<T>::value>::type* = nullptr>
    void scan(T& a) {
        skip_space();
        bool neg = false;
        if (cur() == '-') {
            neg = true;
            next();
        }
        a = 0;
        while ('0' <= cur() && cur() <= '9') {
            a = a * 10 + (scan_char() & 15);
        }
        if (cur() == '.') {
            next();
            T n = 0, d = 1;
            for (int i = 0; '0' <= cur() && cur() <= '9' && i < decimal_precision; ++i) {
                n = n * 10 + (scan_char() & 15);
                d *= 10;
            }
            while ('0' <= cur() && cur() <= '9') next();
            a += n / d;
        }
        if (neg) a = -a;
    }

   private:
    template <std::size_t i, typename... Args>
    void scan(std::tuple<Args...>& a) {
        if constexpr (i < sizeof...(Args)) {
            scan(std::get<i>(a));
            scan<i + 1, Args...>(a);
        }
    }

   public:
    template <typename... Args>
    void scan(std::tuple<Args...>& a) {
        scan<0, Args...>(a);
    }
    template <typename T, typename U>
    void scan(std::pair<T, U>& a) {
        scan(a.first);
        scan(a.second);
    }
    template <typename T, typename std::enable_if<is_range<T>::value && !has_scan<T>::value>::type* = nullptr>
    void scan(T& a) {
        for (auto& i : a) scan(i);
    }
    template <typename T, typename std::enable_if<has_scan<T>::value>::type* = nullptr>
    void scan(T& a) {
        a.scan(*this);
    }
    void operator()() {}
    template <typename Head, typename... Tail>
    void operator()(Head& head, Tail&... tail) {
        scan(head);
        operator()(std::forward<Tail&>(tail)...);
    }
    template <typename T>
    Scanner& operator>>(T& a) {
        scan(a);
        return *this;
    }
    explicit operator bool() const { return state; }
    friend Scanner& getline(Scanner& sc, std::string& a) {
        a.clear();
        char c;
        if ((c = sc.scan_char()) == '\0' || c == '\n') return sc;
        a += c;
        while ((c = sc.scan_char()) != '\0' && c != '\n') a += c;
        return sc;
    }
};
Scanner<> sc;
}  // namespace fastio
using fastio::sc;
#line 6 "template/out.hpp"

namespace fastio {
struct Pre {
    char buffer[10000][4];
    constexpr Pre() : buffer() {
        for (int i = 0; i < 10000; ++i) {
            int n = i;
            for (int j = 3; j >= 0; --j) {
                buffer[i][j] = n % 10 | '0';
                n /= 10;
            }
        }
    }
} constexpr pre;
template <std::size_t BUFF_SIZE = 1 << 17, bool debug = false>
struct Printer {
   private:
    template <typename, bool = debug, class = void>
    struct has_print : std::false_type {};
    template <typename T>
    struct has_print<T, false, decltype(std::declval<T>().print(std::declval<Printer&>()), (void)0)> : std::true_type {};
    template <typename T>
    struct has_print<T, true, decltype(std::declval<T>().debug(std::declval<Printer&>()), (void)0)> : std::true_type {};
    int fd;
    char buffer[BUFF_SIZE];
    int idx;
    std::size_t decimal_precision;

   public:
    Printer() : Printer((debug ? 2 : 1)) {}
    explicit Printer(int fd) : fd(fd), idx(0), decimal_precision(16) {}
    explicit Printer(FILE* file) : fd(fileno(file)), idx(0), decimal_precision(16) {}
    ~Printer() {
        flush();
    }
    void set_decimal_precision(std::size_t n) { decimal_precision = n; }
    inline void print_char(char c) {
        buffer[idx++] = c;
        if (idx == BUFF_SIZE) flush();
    }
    inline void flush() {
        idx = write(fd, buffer, idx);
        idx = 0;
    }
    void print(char a) {
        if constexpr (debug) print_char('\'');
        print_char(a);
        if constexpr (debug) print_char('\'');
    }
    void print(bool a) {
        if constexpr (debug) print_char('\'');
        print_char('0' + a);
        if constexpr (debug) print_char('\'');
    }
    void print(const char* a) {
        if constexpr (debug) print_char('\"');
        for (; *a != '\0'; ++a) print_char(*a);
        if constexpr (debug) print_char('\"');
    }
    template <std::size_t N>
    void print(const char (&a)[N]) {
        if constexpr (debug) print_char('\"');
        for (auto i : a) print_char(i);
        if constexpr (debug) print_char('\"');
    }
    void print(const std::string& a) {
        if constexpr (debug) print_char('\"');
        for (auto i : a) print_char(i);
        if constexpr (debug) print_char('\"');
    }
    template <std::size_t len>
    void print(const std::bitset<len>& a) {
        for (int i = len - 1; i >= 0; --i) print_char('0' + a[i]);
    }
    template <typename T, typename std::enable_if<is_int<T>::value && !has_print<T>::value>::type* = nullptr>
    void print(T a) {
        if (!a) {
            print_char('0');
            return;
        }
        if constexpr (is_signed_int<T>::value) {
            if (a < 0) {
                print_char('-');
                a = -a;
            }
        }
        if (static_cast<size_t>(idx + 40) >= BUFF_SIZE) flush();
        static char stk[40];
        int top = 40;
        while (a >= 10000) {
            int i = a % 10000;
            a /= 10000;
            top -= 4;
            std::memcpy(stk + top, pre.buffer[i], 4);
        }
        if (a >= 1000) {
            std::memcpy(buffer + idx, pre.buffer[a], 4);
            idx += 4;
        } else if (a >= 100) {
            std::memcpy(buffer + idx, pre.buffer[a] + 1, 3);
            idx += 3;
        } else if (a >= 10) {
            std::memcpy(buffer + idx, pre.buffer[a] + 2, 2);
            idx += 2;
        } else {
            buffer[idx++] = '0' | a;
        }
        std::memcpy(buffer + idx, stk + top, 40 - top);
        idx += 40 - top;
    }
    template <typename T, typename std::enable_if<std::is_floating_point<T>::value && !has_print<T>::value>::type* = nullptr>
    void print(T a) {
        if (a == infinity<T>::max || a == infinity<T>::value) {
            print("inf");
            return;
        }
        if (a == infinity<T>::min || a == infinity<T>::mvalue) {
            print("-inf");
            return;
        }
        if (std::isnan(a)) {
            print("nan");
            return;
        }
        if (a < 0) {
            print_char('-');
            a = -a;
        }
        T b = a;
        if (b < 1) {
            print_char('0');
        } else {
            std::string s;
            while (b >= 1) {
                s += (char)('0' | (int)std::fmod(b, 10.0));
                b /= 10;
            }
            for (auto i = s.rbegin(); i != s.rend(); ++i) {
                print_char(*i);
            }
        }
        print_char('.');
        for (std::size_t _ = 0; _ < decimal_precision; ++_) {
            a *= 10;
            print_char('0' | (int)std::fmod(a, 10.0));
        }
    }

   private:
    template <std::size_t i, typename... Args>
    void print(const std::tuple<Args...>& a) {
        if constexpr (i < sizeof...(Args)) {
            if constexpr (debug) print_char(',');
            print_char(' ');
            print(std::get<i>(a));
            print<i + 1>(a);
        }
    }

   public:
    template <typename... Args>
    void print(const std::tuple<Args...>& a) {
        if constexpr (debug) print_char('(');
        if constexpr (sizeof...(Args) != 0) {
            print(std::get<0>(a));
        }
        print<1, Args...>(a);
        if constexpr (debug) print_char(')');
    }
    template <typename T, typename U>
    void print(const std::pair<T, U>& a) {
        if constexpr (debug) print_char('(');
        print(a.first);
        if constexpr (debug) print_char(',');
        print_char(' ');
        print(a.second);
        if constexpr (debug) print_char(')');
    }
    template <typename T, typename std::enable_if<is_range<T>::value>::type* = nullptr>
    void print(const T& a) {
        if constexpr (debug) print_char('{');
        auto it = std::begin(a);
        if (it != std::end(a)) {
            print(*it);
            while (++it != std::end(a)) {
                if constexpr (debug) print_char(',');
                print_char(' ');
                print(*it);
            }
        }
        if constexpr (debug) print_char('}');
    }
    template <typename T, typename std::enable_if<has_print<T>::value && !debug>::type* = nullptr>
    void print(const T& a) {
        a.print(*this);
    }
    template <typename T, typename std::enable_if<has_print<T>::value && debug>::type* = nullptr>
    void print(const T& a) {
        a.debug(*this);
    }
    void operator()() {}
    template <typename Head, typename... Tail>
    void operator()(const Head& head, const Tail&... tail) {
        print(head);
        operator()(std::forward<const Tail&>(tail)...);
    }
    template <typename T>
    Printer& operator<<(const T& a) {
        print(a);
        return *this;
    }
    Printer& operator<<(Printer& (*f)(Printer&)) {
        return f(*this);
    }
};

template <std::size_t BUFF_SIZE, bool debug>
Printer<BUFF_SIZE, debug>& endl(Printer<BUFF_SIZE, debug>& out) {
    out.print_char('\n');
    out.flush();
    return out;
}
template <std::size_t BUFF_SIZE, bool debug>
Printer<BUFF_SIZE, debug>& flush(Printer<BUFF_SIZE, debug>& out) {
    out.flush();
    return out;
}
Printer<> pr;
Printer<1 << 17, true> prd;
}  // namespace fastio
using fastio::endl;
using fastio::flush;
using fastio::pr;
using fastio::prd;
#line 3 "template/func.hpp"

#line 8 "template/func.hpp"

inline constexpr int msb(ull x) {
    int res = x ? 0 : -1;
    if (x & 0xffffffff00000000) x &= 0xffffffff00000000, res += 32;
    if (x & 0xffff0000ffff0000) x &= 0xffff0000ffff0000, res += 16;
    if (x & 0xff00ff00ff00ff00) x &= 0xff00ff00ff00ff00, res += 8;
    if (x & 0xf0f0f0f0f0f0f0f0) x &= 0xf0f0f0f0f0f0f0f0, res += 4;
    if (x & 0xcccccccccccccccc) x &= 0xcccccccccccccccc, res += 2;
    return res + (x & 0xaaaaaaaaaaaaaaaa ? 1 : 0);
}
inline constexpr int ceil_log2(ull x) { return x ? msb(x - 1) + 1 : 0; }
inline constexpr ull reverse(ull x) {
    x = ((x & 0x5555555555555555) << 1) | ((x & 0xaaaaaaaaaaaaaaaa) >> 1);
    x = ((x & 0x3333333333333333) << 2) | ((x & 0xcccccccccccccccc) >> 2);
    x = ((x & 0x0f0f0f0f0f0f0f0f) << 4) | ((x & 0xf0f0f0f0f0f0f0f0) >> 4);
    x = ((x & 0x00ff00ff00ff00ff) << 8) | ((x & 0xff00ff00ff00ff00) >> 8);

    x = ((x & 0x0000ffff0000ffff) << 16) | ((x & 0xffff0000ffff0000) >> 16);
    return (x << 32) | (x >> 32);
}
inline constexpr ull reverse(ull x, int len) { return reverse(x) >> (64 - len); }
inline constexpr int popcnt(ull x) {
#if __cplusplus >= 202002L
    return std::popcount(x);
#endif
    x = (x & 0x5555555555555555) + ((x >> 1) & 0x5555555555555555);
    x = (x & 0x3333333333333333) + ((x >> 2) & 0x3333333333333333);
    x = (x & 0x0f0f0f0f0f0f0f0f) + ((x >> 4) & 0x0f0f0f0f0f0f0f0f);
    x = (x & 0x00ff00ff00ff00ff) + ((x >> 8) & 0x00ff00ff00ff00ff);
    x = (x & 0x0000ffff0000ffff) + ((x >> 16) & 0x0000ffff0000ffff);
    return (x & 0x00000000ffffffff) + ((x >> 32) & 0x00000000ffffffff);
}
template <typename T, typename U>
inline constexpr bool chmin(T& a, U b) { return a > b && (a = b, true); }
template <typename T, typename U>
inline constexpr bool chmax(T& a, U b) { return a < b && (a = b, true); }
inline constexpr ll gcd(ll a, ll b) {
    if (a < 0) a = -a;
    if (b < 0) b = -b;
    while (b) {
        const ll c = b;
        b = a % b;
        a = c;
    }
    return a;
}
inline constexpr ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
inline constexpr bool is_prime(ll n) {
    if (n <= 1) return false;
    for (ll i = 2; i * i <= n; i++) {
        if (n % i == 0) return false;
    }
    return true;
}
inline constexpr ll my_pow(ll a, ll b) {
    ll res = 1;
    while (b) {
        if (b & 1) res *= a;
        a *= a;
        b >>= 1;
    }
    return res;
}
inline constexpr ll mod_pow(ll a, ll b, const ll& mod) {
    if (mod == 1) return 0;
    a %= mod;
    ll res = 1;
    while (b) {
        if (b & 1) (res *= a) %= mod;
        (a *= a) %= mod;
        b >>= 1;
    }
    return res;
}
inline ll mod_inv(ll a, const ll& mod) {
    ll b = mod, x = 1, u = 0, t;
    while (b) {
        t = a / b;
        std::swap(a -= t * b, b);
        std::swap(x -= t * u, u);
    }
    if (x < 0) x += mod;
    return x;
}
template <typename T, typename U>
std::ostream& operator<<(std::ostream& os, const std::pair<T, U>& p) {
    os << p.first << " " << p.second;
    return os;
}
template <typename T, typename U>
std::istream& operator>>(std::istream& is, std::pair<T, U>& p) {
    is >> p.first >> p.second;
    return is;
}
template <typename T>
std::ostream& operator<<(std::ostream& os, const std::vector<T>& v) {
    for (auto it = std::begin(v); it != std::end(v);) {
        os << *it << ((++it) != std::end(v) ? " " : "");
    }
    return os;
}
template <typename T>
std::istream& operator>>(std::istream& is, std::vector<T>& v) {
    for (T& in : v) {
        is >> in;
    }
    return is;
}
inline void scan() {}
template <class Head, class... Tail>
inline void scan(Head& head, Tail&... tail) {
    sc >> head;
    scan(tail...);
}
template <class T>
inline void print(const T& t) { pr << t << '\n'; }
template <class Head, class... Tail>
inline void print(const Head& head, const Tail&... tail) {
    pr << head << ' ';
    print(tail...);
}
template <class... T>
inline void fin(const T&... a) {
    print(a...);
    exit(0);
}
template <typename T>
inline void dump(const T& a) { prd << a; }
inline void trace() { prd << endl; }
template <typename Head, typename... Tail>
inline void trace(const Head& head, const Tail&... tail) {
    dump(head);
    if (sizeof...(tail)) prd.print_char(','), prd.print_char(' ');
    trace(tail...);
}
#ifdef ONLINE_JUDGE
#define dbg(...) (void(0))
#else
#define dbg(...)                                                       \
    do {                                                               \
        prd << #__VA_ARGS__;                                           \
        prd.print_char(' '), prd.print_char('='), prd.print_char(' '); \
        trace(__VA_ARGS__);                                            \
    } while (0)
#endif
#line 3 "template/util.hpp"

#line 6 "template/util.hpp"
template <typename F>
struct REC {
   private:
    F f;

   public:
    explicit constexpr REC(F&& f_) : f(std::forward<F>(f_)) {}
    template <typename... Args>
    constexpr auto operator()(Args&&... args) const {
        return f(*this, std::forward<Args>(args)...);
    }
};
template <typename T, typename Comp = std::less<T>>
struct compressor {
   private:
    std::vector<T> data;
    Comp cmp;
    bool sorted = false;

   public:
    compressor() : compressor(Comp()) {}
    compressor(const Comp& cmp) : cmp(cmp) {}
    compressor(const std::vector<T>& dat, const Comp& cmp = Comp()) : data(dat), cmp(cmp) {}
    compressor(std::vector<T>&& dat, const Comp& cmp = Comp()) : data(std::move(dat)), cmp(cmp) {}
    compressor(std::initializer_list<T> li, const Comp& cmp = Comp()) : data(li.begin(), li.end()), cmp(cmp) {}
    void push_back(const T& v) {
        assert(!sorted);
        data.push_back(v);
    }
    void push_back(T&& v) {
        assert(!sorted);
        data.push_back(std::move(v));
    }
    template <typename... Args>
    void emplace_back(Args&&... args) {
        assert(!sorted);
        data.emplace_back(std::forward<Args>(args)...);
    }
    void push(const std::vector<T>& v) {
        assert(!sorted);
        const int n = data.size();
        data.resize(v.size() + n);
        for (int i = 0; i < (int)v.size(); i++) data[i + n] = v[i];
    }
    void build() {
        assert(!sorted);
        sorted = 1;
        std::sort(data.begin(), data.end(), cmp);
        data.erase(unique(data.begin(), data.end(), [&](const T& l, const T& r) -> bool { return !cmp(l, r) && !cmp(r, l); }), data.end());
    }
    const T& operator[](int k) const& {
        assert(sorted);
        return data[k];
    }
    int get_index(const T& v) const {
        assert(sorted);
        return int(lower_bound(data.begin(), data.end(), v, cmp) - data.begin());
    }
    void press(std::vector<T>& v) const {
        assert(sorted);
        for (auto&& i : v) i = get_index(i);
    }
    std::vector<int> pressed(const std::vector<T>& v) const {
        assert(sorted);
        std::vector<int> ret(v.size());
        for (int i = 0; i < (int)v.size(); i++) ret[i] = get_index(v[i]);
        return ret;
    }
    int size() const {
        assert(sorted);
        return data.size();
    }
};
#line 11 "template/template.hpp"
using namespace std;
#line 3 "ds/others/hash-map.hpp"

template <typename Key, typename Val>
struct HashMap {
    using u32 = uint32_t;
    using u64 = uint64_t;
    using Data = pair<Key, Val>;

   protected:
    template <typename K>
    inline u64 randomized(const K& key) const {
        return u64(key) ^ r;
    }
    template <typename K, enable_if_t<is_integral<K>::value, nullptr_t> = nullptr>
    inline u64 inner_hash(const K& key) const {
        return (randomized(key) * 11995408973635179863ULL);
    }
    template <typename K, enable_if_t<is_integral<decltype(K::first)>::value, nullptr_t> = nullptr, enable_if_t<is_integral<decltype(K::second)>::value, nullptr_t> = nullptr>
    inline u64 inner_hash(const K& key) const {
        u64 a = randomized(key.first), b = randomized(key.second);
        a *= 11995408973635179863ULL;
        b *= 10150724397891781847ULL;
        return (a + b);
    }
    template <typename K, enable_if_t<is_integral<typename K::value_type>::value, nullptr_t> = nullptr>
    inline u64 inner_hash(const K& key) const {
        static constexpr u64 mod = (1LL << 61) - 1;
        static constexpr u64 base = 950699498548472943ULL;
        u64 res = 0;
        for (auto& elem : key) {
            __uint128_t x = __uint128_t(res) * base + (randomized(elem) & mod);
            res = (x & mod) + (x >> 61);
        }
        __uint128_t x = __uint128_t(res) * base;
        res = (x & mod) + (x >> 61);
        if (res >= mod) res -= mod;
        return (res << 3);
    }
    inline u32 hash(const Key& key) const {
        return inner_hash(key) >> shift;
    }
    void reallocate(u32 new_cap) {
        vector<Data> new_data(new_cap);
        vector<bool> new_flag(new_cap, false);
        shift = 64 - __lg(new_cap);
        for (u32 i = 0; i < cap; i++) {
            if (flag[i] && !dflag[i]) {
                u32 h = hash(data[i].first);
                while (new_flag[h]) h = (h + 1) & (new_cap - 1);
                new_data[h] = move(data[i]);
                new_flag[h] = true;
            }
        }
        data.swap(new_data);
        flag.swap(new_flag);
        cap = new_cap;
        dflag.resize(cap);
        fill(dflag.begin(), dflag.end(), false);
    }
    inline bool should_extend(u32 x) const { return x * 2 >= cap; }
    inline bool should_shrink(u32 x) const { return 4 < cap && x * 10 <= cap; }
    inline void extend() { reallocate(cap << 1); }
    inline void shrink() { reallocate(cap >> 1); }

   public:
    u32 cap, s;
    vector<Data> data;
    vector<bool> flag, dflag;
    u32 shift;
    static u64 r;
    static constexpr uint32_t DEFAULT_SIZE = 4;
    struct iterator {
        u32 i;
        HashMap<Key, Val>* p;
        explicit constexpr iterator() : i(0), p(nullptr) {}
        explicit constexpr iterator(u32 i, HashMap<Key, Val>* p) : i(i), p(p) {}
        explicit constexpr iterator(u32 i, const HashMap<Key, Val>* p) : i(i), p(const_cast<HashMap<Key, Val>*>(p)) {}
        const Data& operator*() const {
            return const_cast<HashMap<Key, Val>*>(p)->data[i];
        }
        Data& operator*() { return p->data[i]; }
        Data* operator->() { return &(p->data[i]); }
        friend void swap(iterator& a, iterator& b) {
            swap(a.i, b.i);
            swap(a.p, b.p);
        }
        friend bool operator==(const iterator& a, const iterator& b) { return a.i == b.i; }
        friend bool operator!=(const iterator& a, const iterator& b) { return a.i != b.i; }
        iterator& operator++() {
            assert(i != p->cap && "iterator overflow");
            do {
                i++;
                if (i == p->cap) break;
                if (p->flag[i] && !(p->dflag[i])) break;
            } while (true);
            return *this;
        }
        iterator operator++(int) {
            iterator tmp(*this);
            ++(*this);
            return tmp;
        }
        iterator& operator--() {
            do {
                i--;
                if (p->flag[i] && !(p->dflag[i])) break;
                assert(i != 0 && "iterator underflow");
            } while (true);
            return *this;
        }
        iterator operator--(int) {
            iterator tmp(*this);
            --(*this);
            return tmp;
        }
    };
    using itr = iterator;
    explicit HashMap() : cap(DEFAULT_SIZE), s(0), data(cap), flag(cap), dflag(cap), shift(62) {}
    itr begin() const {
        u32 h = 0;
        while (h != cap) {
            if (flag[h] && !dflag[h]) break;
            h++;
        }
        return itr(h, this);
    }
    itr end() const { return itr(this->cap, this); }
    friend itr begin(HashMap<Key, Val>& a) { return a.begin(); }
    friend itr end(HashMap<Key, Val>& a) { return a.end(); }
    itr find(const Key& key) const {
        u32 h = hash(key);
        while (true) {
            if (!flag[h]) return this->end();
            if (data[h].first == key) {
                if (dflag[h]) return this->end();
                return itr(h, this);
            }
            h = (h + 1) & (cap - 1);
        }
    }
    bool contain(const Key& key) const { return find(key) != this->end(); }
    int count(const Key& key) const { return int(find(key) != this->end()); }
    itr insert(const Data& d) {
        u32 h = hash(d.first);
        while (true) {
            if (!flag[h]) {
                if (should_extend(s + 1)) {
                    extend();
                    h = hash(d.first);
                    continue;
                }
                data[h] = d;
                flag[h] = true;
                s++;
                return itr(h, this);
            }
            if (data[h].first == d.first) {
                if (dflag[h]) {
                    data[h] = d;
                    dflag[h] = false;
                    s++;
                }
                return itr(h, this);
            }
            h = (h + 1) & (cap - 1);
        }
    }
    bool erase(itr it) {
        if (it == this->end()) return false;
        s--;
        if (should_shrink(s)) {
            Data d = data[it.i];
            shrink();
            it = find(d.first);
        }
        int ni = (it.i + 1) & (cap - 1);
        if (flag[ni]) {
            dflag[it.i] = true;
        } else {
            flag[it.i] = false;
        }
        return true;
    }
    bool erase(const Key& key) { return erase(find(key)); }
    bool empty() const { return s == 0; }
    u32 size() const { return s; }
    void clear() {
        s = 0;
        fill(flag.begin(), flag.end(), false);
        fill(dflag.begin(), dflag.end(), false);
    }
    void reserve(int n) {
        if (n <= 0) return;
        n = 1 << (__lg(n) + 2);
        if (cap < u32(n)) reallocate(n);
    }
    Val& operator[](const Key& key) {
        u32 h = hash(key);
        while (true) {
            if (!flag[h]) {
                if (should_extend(s + 1)) {
                    extend();
                    h = hash(key);
                    continue;
                }
                data[h] = Data(key, Val());
                flag[h] = true;
                s++;
                return data[h].second;
            }
            if (data[h].first == key) {
                if (dflag[h]) data[h].second = Val();
                return data[h].second;
            }
            h = (h + 1) & (cap - 1);
        }
    }
    bool emplace(const Key& key, const Val& val) {
        return insert(Data(key, val));
    }
};
template <typename Key, typename Val>
uint64_t HashMap<Key, Val>::r = chrono::duration_cast<chrono::nanoseconds>(chrono::system_clock::now().time_since_epoch()).count();
/**
 * @brief HashMap(ハッシュマップ)
 */
#line 4 "math/modular/mod-log.hpp"

template <typename T>
T discrete_logarithm(T x, T y, T m) {
    x %= m, y %= m;
    if (y == 1 || m == 1) return 0;
    if (x == 0) return y == 0 ? 1 : -1;
    T add = 0, g, k = 1 % m;
    while ((g = gcd(x, m)) > 1) {
        if (y == k) return add;
        if (y % g) return -1;
        y /= g, m /= g, add++;
        k = (k * (x / g)) % m;
    }
    T n = sqrt(m) + 1;
    T tmp = mod_pow(x, n, m);
    HashMap<T, T> mp;
    for (T i = 0, now = y; i <= n; i++) {
        mp[now] = i;
        now = (now * x) % m;
    }
    for (T i = 1, now = k; i <= n; i++) {
        now = (now * tmp) % m;
        if (mp.contain(now)) return n * i - mp[now] + add;
    }
    return -1;
}
/**
 * @brief Mod Log(離散対数)
 */
#line 4 "test/yosupo/number_theory/discrete_logarithm_mod.test.cpp"
int main() {
    int t;
    sc >> t;
    rep(t) {
        ll x, y, m;
        sc >> x >> y >> m;
        print(discrete_logarithm(x, y, m));
    }
}

Test cases

Env Name Status Elapsed Memory
g++ even_mod_00 :heavy_check_mark: AC 98 ms 5 MB
g++ even_mod_01 :heavy_check_mark: AC 121 ms 5 MB
g++ even_mod_impossible_00 :heavy_check_mark: AC 6 ms 3 MB
g++ even_mod_impossible_01 :heavy_check_mark: AC 6 ms 3 MB
g++ example_00 :heavy_check_mark: AC 6 ms 3 MB
g++ max_random_00 :heavy_check_mark: AC 148 ms 5 MB
g++ max_random_01 :heavy_check_mark: AC 137 ms 5 MB
g++ max_random_02 :heavy_check_mark: AC 127 ms 5 MB
g++ max_random_yes_00 :heavy_check_mark: AC 140 ms 5 MB
g++ max_random_yes_01 :heavy_check_mark: AC 138 ms 5 MB
g++ max_random_yes_prime_00 :heavy_check_mark: AC 190 ms 5 MB
g++ max_random_yes_prime_01 :heavy_check_mark: AC 194 ms 5 MB
g++ random_00 :heavy_check_mark: AC 46 ms 5 MB
g++ random_01 :heavy_check_mark: AC 74 ms 5 MB
g++ random_02 :heavy_check_mark: AC 97 ms 5 MB
g++ random_prime_00 :heavy_check_mark: AC 204 ms 5 MB
g++ random_prime_01 :heavy_check_mark: AC 203 ms 5 MB
g++ small_00 :heavy_check_mark: AC 6 ms 3 MB
g++ small_01 :heavy_check_mark: AC 6 ms 3 MB
g++ small_02 :heavy_check_mark: AC 6 ms 3 MB
Back to top page