逆元計算の関数の定義. invmod().
(ugilib/math/invmod.hpp)

• View this file on GitHub
• Last update: 2024-04-26 23:52:15+09:00
• Include: #include "ugilib/math/invmod.hpp"

Depends on

• ugilib/base/constants.hpp
• ugilib/base/definitions.hpp

Required by

• 組み合わせ計算の関数の定義. perm(), comb(), fact(). (ugilib/math/combinatorics.hpp)

Code

#pragma once

/**
 * @file invmod.hpp
 * @brief 逆元計算の関数の定義. invmod().
*/

#include <bits/stdc++.h>
#include "ugilib/base/definitions.hpp"
#include "ugilib/base/constants.hpp"

using namespace std;

namespace ugilib {
    /**
     * @brief 拡張ユークリッドの互除法
     * @tparam T 整数型
     * @param a a
     * @param b b
     * @param x x
     * @param y y
     * @return T gcd(a, b)
     * @note O(log^2(max(a, b)))
     * @note ax + by = gcd(a, b)を満たすx, yを求める
    */
    template <typename T>
    T extended_gcd(T a, T b, T& x, T& y) {
        if (b == 0) {
            x = 1;
            y = 0;
            return a;
        }
        T gcd = extended_gcd(b, a % b, x, y);
        T tmp = x;
        x = y;
        y = tmp - a / b * y;
        return gcd;
    }

    /**
     * @brief aのmodにおける逆元を求める
     * @tparam T 整数型
     * @param a a
     * @param mod mod
     * @return T aのmodにおける逆元
     * @note O(log^2(mod))
     * @note modが素数であり、互いに素であることが前提
    */
    template <typename T>
    T invmod(T a, T mod) {
        T x, y;
        T gcd = extended_gcd(a, mod, x, y);
        assert(gcd == 1);
        return (x % mod + mod) % mod;
    }
}

#line 2 "ugilib/math/invmod.hpp"

/**
 * @file invmod.hpp
 * @brief 逆元計算の関数の定義. invmod().
*/

#include <bits/stdc++.h>
#line 2 "ugilib/base/definitions.hpp"

using ll = long long;
using ull = unsigned long long;
using ld = long double;
#define rep(i, n) for(size_t i = 0; i < n; i++)  // rep macro
#define all(v) begin(v), end(v)  // all iterator
#line 3 "ugilib/base/constants.hpp"

namespace ugilib::constants {
    template<typename T>
    inline constexpr T INF = std::numeric_limits<T>::max() / 2;
} // namespace ugilib::constants

const ll INF = ugilib::constants::INF<ll>;
#line 11 "ugilib/math/invmod.hpp"

using namespace std;

namespace ugilib {
    /**
     * @brief 拡張ユークリッドの互除法
     * @tparam T 整数型
     * @param a a
     * @param b b
     * @param x x
     * @param y y
     * @return T gcd(a, b)
     * @note O(log^2(max(a, b)))
     * @note ax + by = gcd(a, b)を満たすx, yを求める
    */
    template <typename T>
    T extended_gcd(T a, T b, T& x, T& y) {
        if (b == 0) {
            x = 1;
            y = 0;
            return a;
        }
        T gcd = extended_gcd(b, a % b, x, y);
        T tmp = x;
        x = y;
        y = tmp - a / b * y;
        return gcd;
    }

    /**
     * @brief aのmodにおける逆元を求める
     * @tparam T 整数型
     * @param a a
     * @param mod mod
     * @return T aのmodにおける逆元
     * @note O(log^2(mod))
     * @note modが素数であり、互いに素であることが前提
    */
    template <typename T>
    T invmod(T a, T mod) {
        T x, y;
        T gcd = extended_gcd(a, mod, x, y);
        assert(gcd == 1);
        return (x % mod + mod) % mod;
    }
}

Back to top page