#line 1 "Library/src/debug.hpp" #ifdef ONLINE_JUDGE #define debug(x) void(0) #else #define _GLIBCXX_DEBUG #define debug(x) std::cerr << __LINE__ << " : " << #x << " = " << (x) << std::endl #endif /** * @brief Debugger */ #line 2 "Library/src/math/rho.hpp" #include #include #line 2 "Library/src/math/gcd.hpp" #include #include #include namespace kyopro { template constexpr inline T _gcd(T a, T b) noexcept { assert(a >= 0 && b >= 0); if (a == 0 || b == 0) return a + b; int d = std::min(__builtin_ctzll(a), __builtin_ctzll(b)); a >>= __builtin_ctzll(a), b >>= __builtin_ctzll(b); while (a != b) { if (!a || !b) { return a + b; } if (a >= b) { a -= b; a >>= __builtin_ctzll(a); } else { b -= a; b >>= __builtin_ctzll(b); } } return a << d; } template constexpr inline T ext_gcd(T a, T b, T& x, T& y) noexcept { x = 1, y = 0; T nx = 0, ny = 1; while (b) { T q = a / b; std::tie(a, b) = std::pair{b, a % b}; std::tie(x, nx) = std::pair{nx, x - nx * q}; std::tie(y, ny) = std::pair{ny, y - ny * q}; } return a; } }; // namespace kyopro /** * @brief gcd */ #line 3 "Library/src/math/dynamic_modint.hpp" #include #line 2 "Library/src/internal/barrett.hpp" #include namespace kyopro { namespace internal { class barrett { using u32 = std::uint32_t; using u64 = std::uint64_t; using u128 = __uint128_t; u32 m; u64 im; public: constexpr barrett() : m(0), im(0) {} constexpr barrett(u32 m) : m(m), im(static_cast(-1) / m + 1) {} constexpr u32 get_mod() const { return m; } constexpr u32 reduce(u32 a) const { return mul(1, a); } constexpr u32 mul(u32 a, u32 b) const { u64 z = (u64)a * b; u64 x = (u64)(((u128)(z)*im) >> 64); u64 y = x * m; return (u32)(z - y + (z < y ? m : 0)); } }; }; // namespace internal }; // namespace kyopro /** * @brief Barrett Reduction * @see https://github.com/atcoder/ac-library/blob/master/atcoder/internal_math.hpp */ #line 3 "Library/src/internal/montgomery.hpp" #include #include #line 5 "Library/src/internal/type_traits.hpp" #include #line 7 "Library/src/internal/type_traits.hpp" namespace kyopro { namespace internal { template struct first_enabled {}; template struct first_enabled, Args...> { using type = T; }; template struct first_enabled, Args...> : first_enabled {}; template struct first_enabled { using type = T; }; template using first_enabled_t = typename first_enabled::type; template * = nullptr> struct int_least { using type = first_enabled_t, std::enable_if, std::enable_if, std::enable_if, std::enable_if>; }; template * = nullptr> struct uint_least { using type = first_enabled_t, std::enable_if, std::enable_if, std::enable_if, std::enable_if>; }; template using int_least_t = typename int_least::type; template using uint_least_t = typename uint_least::type; template using double_size_uint_t = uint_least_t<2 * std::numeric_limits::digits>; template using double_size_int_t = int_least_t<2 * std::numeric_limits::digits>; struct modint_base {}; template using is_modint = std::is_base_of; template using is_modint_t = std::enable_if_t::value>; // is_integral template using is_integral_t = std::enable_if_t || std::is_same_v || std::is_same_v>; }; // namespace internal }; // namespace kyopro /** * @brief Type Traits * @see https://qiita.com/kazatsuyu/items/f8c3b304e7f8b35263d8 */ #line 6 "Library/src/internal/montgomery.hpp" namespace kyopro { namespace internal { using u32 = uint32_t; using u64 = uint64_t; using i32 = int32_t; using i64 = int64_t; using u128 = __uint128_t; using i128 = __int128_t; template class Montgomery { static constexpr int lg = std::numeric_limits::digits; using LargeT = internal::double_size_uint_t; T mod, r, r2, minv; T inv() { T t = 0, res = 0; for (int i = 0; i < lg; ++i) { if (~t & 1) { t += mod; res += static_cast(1) << i; } t >>= 1; } return res; } public: Montgomery() = default; constexpr T get_mod() { return mod; } void set_mod(T m) { assert(m); assert(m & 1); mod = m; r = (-static_cast(mod)) % mod; r2 = (-static_cast(mod)) % mod; minv = inv(); } T reduce(LargeT x) const { u64 res = (x + static_cast(static_cast(x) * minv) * mod) >> lg; if (res >= mod) res -= mod; return res; } T generate(LargeT x) { return reduce(x * r2); } T mul(T x, T y) { return reduce((LargeT)x * y); } }; }; // namespace internal }; // namespace kyopro /** * @brief Montgomery Reduction */ #line 6 "Library/src/math/dynamic_modint.hpp" namespace kyopro { template class barrett_modint : internal::modint_base { using mint = barrett_modint; using u32 = std::uint32_t; using u64 = std::uint64_t; using i32 = std::int32_t; using i64 = std::int64_t; using br = internal::barrett; static br brt; u32 v; public: static void set_mod(u32 mod_) { brt = br(mod_); } public: explicit constexpr barrett_modint() noexcept : v(0) { assert(mod()); } explicit constexpr barrett_modint(i64 v_) noexcept : v() { assert(mod()); if (v_ < 0) v_ = (i64)mod() - v_; v = brt.reduce(v_); } u32 val() const noexcept { return v; } static u32 mod() { return brt.get_mod(); } static mint raw(u32 v) { mint x; x.v = v; return x; } constexpr mint& operator++() noexcept { ++v; if (v == mod()) v = 0; return (*this); } constexpr mint& operator--() noexcept { if (v == 0) v = mod(); --v; return (*this); } constexpr mint operator++(int) noexcept { mint res(*this); ++(*this); return res; } constexpr mint operator--(int) noexcept { mint res(*this); --(*this); return res; } constexpr mint& operator+=(const mint& r) noexcept { v += r.v; if (v >= mod()) v -= mod(); return (*this); } constexpr mint& operator-=(const mint& r) noexcept { v += mod() - r.v; if (v >= mod()) { v -= mod(); } return (*this); } constexpr mint& operator*=(const mint& r) noexcept { v = brt.mul(v, r.v); return (*this); } constexpr mint& operator/=(const mint& r) noexcept { return (*this) *= r.inv(); } friend mint operator+(const mint& lhs, const mint& rhs) noexcept { return mint(lhs) += rhs; } friend mint operator-(const mint& lhs, const mint& rhs) noexcept { return mint(lhs) -= rhs; } friend mint operator*(const mint& lhs, const mint& rhs) noexcept { return mint(lhs) *= rhs; } friend mint operator/(const mint& lhs, const mint& rhs) noexcept { return mint(lhs) /= rhs; } friend bool operator==(const mint& lhs, const mint& rhs) noexcept { return lhs.v == rhs.v; } friend bool operator!=(const mint& lhs, const mint& rhs) noexcept { return lhs.v != rhs.v; } constexpr mint& operator+=(i64 r) noexcept { return (*this) += mint(r); } constexpr mint& operator-=(i64 r) noexcept { return (*this) -= mint(r); } constexpr mint& operator*=(i64 r) noexcept { return (*this) *= mint(r); } friend mint operator+(i64 l, const mint& r) noexcept { return mint(l) += r; } friend mint operator+(const mint& l, i64 r) noexcept { return mint(l) += r; } friend mint operator-(i64 l, const mint& r) noexcept { return mint(l) -= r; } friend mint operator-(const mint& l, i64 r) noexcept { return mint(l) -= r; } friend mint operator*(i64 l, const mint& r) noexcept { return mint(l) *= r; } friend mint operator*(const mint& l, i64 r) noexcept { return mint(l) *= r; } constexpr mint operator+() const noexcept { return *this; } constexpr mint operator-() const noexcept { return mint() - *this; } template mint pow(T e) const noexcept { mint res(1), base(*this); while (e) { if (e & 1) { res *= base; } e >>= 1; base *= base; } return res; } constexpr mint inv() const noexcept { return pow(mod() - 2); } }; }; // namespace kyopro template typename kyopro::barrett_modint::br kyopro::barrett_modint::brt; namespace kyopro { template class montgomery_modint : internal::modint_base { using LargeT = internal::double_size_uint_t; static T _mod; static internal::Montgomery mr; public: static void set_mod(T mod_) { mr.set_mod(mod_); _mod = mod_; } static T mod() { return _mod; } private: T v; public: montgomery_modint(T v_ = 0) { assert(_mod); v = mr.generate(v_); } T val() const { return mr.reduce(v); } using mint = montgomery_modint; mint& operator+=(const mint& r) { v += r.v; if (v >= mr.get_mod()) { v -= mr.get_mod(); } return (*this); } mint& operator-=(const mint& r) { v += mr.get_mod() - r.v; if (v >= mr.get_mod) { v -= mr.get_mod(); } return (*this); } mint& operator*=(const mint& r) { v = mr.mul(v, r.v); return (*this); } mint operator+(const mint& r) { return mint(*this) += r; } mint operator-(const mint& r) { return mint(*this) -= r; } mint operator*(const mint& r) { return mint(*this) *= r; } mint& operator=(const T& v_) { (*this) = mint(v_); return (*this); } friend std::ostream& operator<<(std::ostream& os, const mint& mt) { os << mt.val(); return os; } friend std::istream& operator>>(std::istream& is, mint& mt) { T v_; is >> v_; mt = v_; return is; } template mint pow(P e) const { assert(e >= 0); mint res(1), base(*this); while (e) { if (e & 1) { res *= base; } e >>= 1; base *= base; } return res; } mint inv() const { return pow(mod() - 2); } mint& operator/=(const mint& r) { return (*this) *= r.inv(); } mint operator/(const mint& r) const { return mint(*this) *= r.inv(); } mint& operator/=(T r) { return (*this) /= mint(r); } friend mint operator/(const mint& l, T r) { return mint(l) /= r; } friend mint operator/(T l, const mint& r) { return mint(l) /= r; } }; }; // namespace kyopro template T kyopro::montgomery_modint::_mod; template kyopro::internal::Montgomery kyopro::montgomery_modint::mr; /** * @brief Dynamic modint */ #line 3 "Library/src/math/miller.hpp" namespace kyopro { class miller { using i128 = __int128_t; using u128 = __uint128_t; using u64 = std::uint64_t; using u32 = std::uint32_t; template static constexpr bool miller_rabin(T n) { T d = n - 1; while (~d & 1) { d >>= 1; } const T rev = n - 1; if (mint::mod() != n) { mint::set_mod(n); } for (int i = 0; i < length; ++i) { if (n <= bases[i]) { return true; } T t = d; mint y = mint(bases[i]).pow(t); while (t != n - 1 && y.val() != 1 && y.val() != rev) { y *= y; t <<= 1; } if (y.val() != rev && (~t & 1)) return false; } return true; } // 底 static constexpr int bases_int[3] = {2, 7, 61}; static constexpr int bases_ll[7] = {2, 325, 9375, 28178, 450775, 9780504, 1795265022}; public: template static constexpr bool is_prime(T n) { if (n < 2) { return false; } else if (n == 2) { return true; } else if (~n & 1) { return false; }; if constexpr (std::numeric_limits::digits < 32) { return miller_rabin>, bases_int, 3>(n); } else { if (n <= 1 << 30) return miller_rabin>, bases_int, 3>(n); else return miller_rabin< T, montgomery_modint>, bases_ll, 7>( n); } return false; } }; }; // namespace kyopro /** * @brief Primality Test(MillerRabin素数判定) * @docs docs/math/miller.md */ #line 2 "Library/src/random/xor_shift.hpp" #include #line 4 "Library/src/random/xor_shift.hpp" #include namespace kyopro { struct xor_shift32 { uint32_t rng; constexpr explicit xor_shift32(uint32_t seed) : rng(seed) {} explicit xor_shift32() : rng(std::chrono::steady_clock::now().time_since_epoch().count()) {} constexpr uint32_t operator()() { rng ^= rng << 13; rng ^= rng >> 17; rng ^= rng << 5; return rng; } }; struct xor_shift { uint64_t rng; constexpr explicit xor_shift(uint64_t seed) : rng(seed) {} explicit xor_shift() : rng(std::chrono::steady_clock::now().time_since_epoch().count()) {} constexpr uint64_t operator()() { rng ^= rng << 13; rng ^= rng >> 7; rng ^= rng << 17; return rng; } }; }; // namespace kyopro /** * @brief Xor Shift */ #line 7 "Library/src/math/rho.hpp" namespace kyopro { class rho { using i128 = __int128_t; using u128 = __uint128_t; using u64 = uint64_t; using u32 = uint32_t; template static constexpr T find_factor(T n) { xor_shift32 rng(2023); if (~n & 1uL) { return 2; } if (kyopro::miller::is_prime(n)) { return n; } if (mint::mod() != n) { mint::set_mod(n); } while (1) { T c = rng(); const auto f = [&](mint x) -> mint { return x * x + c; }; mint x = rng(); mint y = f(x); T d = 1; while (d == 1) { d = _gcd>( std::abs((std::make_signed_t)x.val() - (std::make_signed_t)y.val()), n); x = f(x); y = f(f(y)); } if (1 < d && d < n) { return d; } } exit(-1); } template static std::vector rho_fact(T n) { if (n < 2) { return {}; } if (kyopro::miller::is_prime(n)) { return {n}; } std::vector v; std::vector st{n}; while (!st.empty()) { u64 m = st.back(); if (kyopro::miller::is_prime(m)) { v.emplace_back(m); st.pop_back(); } else { T d = find_factor(m); st.back() /= d; st.emplace_back(d); } } return v; } public: template static std::vector factorize(T n) { if (n < 2) { return {}; } if constexpr (std::numeric_limits::digits < 32) { std::vector v = rho_fact>(n); std::sort(v.begin(), v.end()); return v; } else { std::vector v = rho_fact>(n); std::sort(v.begin(), v.end()); return v; } } template static std::vector> exp_factorize(T n) { std::vector pf = factorize(n); if (pf.empty()) { return {}; } std::vector> res; res.emplace_back(pf.front(), 1); for (int i = 1; i < (int)pf.size(); i++) { if (res.back().first == pf[i]) { res.back().second++; } else { res.emplace_back(pf[i], 1); } } return res; } template static std::vector enumerate_divisor(T n) { std::vector> pf = rho::exp_factorize(n); std::vector divisor{1}; for (auto [p, e] : pf) { u64 pow = p; int sz = divisor.size(); for (int i = 0; i < e; ++i) { for (int j = 0; j < sz; ++j) divisor.emplace_back(divisor[j] * pow); pow *= p; } } return divisor; } }; }; // namespace kyopro /** * @brief PollardRho素因数分解 * @docs docs/math/rho.md */ #line 2 "Library/src/stream.hpp" #include #include #include #line 6 "Library/src/stream.hpp" namespace kyopro { inline void single_read(char& c) { c = getchar_unlocked(); while (isspace(c)) c = getchar_unlocked(); } template * = nullptr> inline void single_read(T& a) { a = 0; bool is_negative = false; char c = getchar_unlocked(); while (isspace(c)) { c = getchar_unlocked(); } if (c == '-') is_negative = true, c = getchar_unlocked(); while (isdigit(c)) { a = 10 * a + (c - '0'); c = getchar_unlocked(); } if (is_negative) a *= -1; } template * = nullptr> inline void single_read(T& a) { long long x; single_read(x); a = T(x); } inline void single_read(std::string& str) noexcept { char c = getchar_unlocked(); while (isspace(c)) c = getchar_unlocked(); while (!isspace(c)) { str += c; c = getchar_unlocked(); } } template inline void read(T& x) noexcept {single_read(x);} template inline void read(Head& head, Tail&... tail) noexcept { single_read(head), read(tail...); } inline void single_write(char c) noexcept { putchar_unlocked(c); } template * = nullptr> inline void single_write(T a) noexcept { if (!a) { putchar_unlocked('0'); return; } if constexpr (std::is_signed_v) { if (a < 0) putchar_unlocked('-'), a *= -1; } constexpr int d = std::numeric_limits::digits10; char s[d + 1]; int now = d + 1; while (a) { s[--now] = (char)'0' + a % 10; a /= 10; } while (now <= d) putchar_unlocked(s[now++]); } template * = nullptr> inline void single_write(T a) noexcept { single_write(a.val()); } inline void single_write(const std::string& str) noexcept { for (auto c : str) { putchar_unlocked(c); } } template inline void write(T x) noexcept { single_write(x); } template inline void write(Head head, Tail... tail) noexcept { single_write(head); putchar_unlocked(' '); write(tail...); } template inline void put(Args... x) noexcept { write(x...); putchar_unlocked('\n'); } }; // namespace kyopro /** * @brief Fast IO(高速入出力) */ #line 2 "Library/src/template.hpp" #include #define rep(i, n) for (int i = 0; i < (n); i++) #define all(x) std::begin(x), std::end(x) #define popcount(x) __builtin_popcountll(x) using i128 = __int128_t; using ll = long long; using ld = long double; using graph = std::vector>; using P = std::pair; constexpr int inf = std::numeric_limits::max() / 2; constexpr ll infl = std::numeric_limits::max() / 2; const long double pi = acosl(-1); constexpr int dx[] = {1, 0, -1, 0, 1, -1, -1, 1, 0}; constexpr int dy[] = {0, 1, 0, -1, 1, 1, -1, -1, 0}; template constexpr inline bool chmax(T1& a, T2 b) { return a < b && (a = b, true); } template constexpr inline bool chmin(T1& a, T2 b) { return a > b && (a = b, true); } /** * @brief Template */ #line 5 "a.cpp" using namespace std; using namespace kyopro; int main() { int n, p; read(n, p); vector a(n); rep(i, n) read(a[i]); ll ans = 0; for (ll q = p; q <= (int)1e9; q *= p) { unordered_map rem; rep(i, n)++ rem[a[i] % q]; for (auto [r, c] : rem) ans += ll(c) * (c - 1) / 2; } put(ans); }