結果
問題 | No.2260 Adic Sum |
ユーザー |
👑 ![]() |
提出日時 | 2023-04-07 21:55:14 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 959 ms / 2,000 ms |
コード長 | 6,988 bytes |
コンパイル時間 | 2,309 ms |
コンパイル使用メモリ | 207,708 KB |
最終ジャッジ日時 | 2025-02-12 01:14:14 |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
#include <bits/stdc++.h>using namespace std;using ll = long long;#ifdef MY_DEBUG#define dbg(...) debug_out(__VA_ARGS__)#define dbgn(var) debug_out(#var, "=", var)#else#define dbg(...)#define dbgn(...)#endifvoid debug_out();template <class Head, class... Tail> void debug_out(Head, Tail...);template <class T> bool chmax(T&, const T&);template <class T> bool chmin(T&, const T&);template <class T> T pwr(T, ll);template <class T> T squ(T x) { return x * x; }ll fact(ll);ll comb(ll, ll);ll ctoll(char);char lltoc(ll);bool flg(ll b, ll i) { return ((b) >> (i)) & 1LL; } // flg(0b010, 1LL) -> trueconst ll LINF = (ll)1e18 + 7;const double EPS = 1e-9;const int MAX_DUBUG_SIZE = 10;// #include <boost/multiprecision/cpp_dec_float.hpp>// #include <boost/multiprecision/cpp_int.hpp>// #include <boost/multiprecision/integer.hpp>// #include <boost/integer/mod_inverse.hpp>// namespace mp = boost::multiprecision;// // 任意長整数型// using Bint = mp::cpp_int;// // 仮数部が10進数で1024桁の浮動小数点数型(TLEしたら小さくする)// using Real = mp::number<mp::cpp_dec_float<1024>>;// // 多倍長整数:https://qiita.com/tubo28/items/fa8ee013390184b0ba18// // mod演算:https://www.k-pmpstudy.com/entry/2019/10/04/moduboost// // mp::powm(ll a, ll p, ll m) = a^p mod m// // boost::integer::mod_inverse(int a, int p) = a^(-1) mod p 逆元が存在するときvoid solve([[maybe_unused]] int test) {ll n, p;cin >> n >> p;vector<ll> a(n);for (ll i = 0; i < n; i++) {cin >> a[i];}ll MAX = 40;vector<map<ll, ll>> cnt(MAX);ll base_p = 1;for (ll i = 1; i < MAX; i++) {if ((long double)base_p * p >= 1'000'000'000) {break;}else {base_p *= p;}for (ll j = 0; j < n; j++) {ll amari = a[j] % base_p;cnt[i][amari]++;}}ll ans = 0;for (auto&& mp : cnt) {for (auto&& [x, y] : mp) {if (y <= 1)continue;else {ans += comb(y, 2);}}}cout << ans << '\n';}int main() {cin.tie(nullptr), ios::sync_with_stdio(false), cout << fixed << setprecision(12);int testcase_size = 1;// cin >> testcase_size;for (int _t = 0; _t < testcase_size; _t++) {solve(_t);}}/*-----------------------------------MY_FUNCTIONS------------------------------------ */template <class First, class Second> ostream& operator<<(ostream& os, const pair<First, Second>& pp) {return os << "{" << pp.first << "," << pp.second << "}";}template <class T> ostream& operator<<(ostream& os, const vector<T>& V) {if (V.empty()) return os << "[]";os << "[";for (ll i = 0; i < V.size(); i++) {os << V[i] << (i == int(V.size() - 1) ? "]" : ",");}return os;}template <class T> ostream& operator<<(ostream& os, const vector<vector<T>>& VV) {if (VV.empty()) return os << "[[]]";os << "[\n";for (auto&& V : VV) {os << V << "\n";}os << "]";return os;}template <class T> ostream& operator<<(ostream& os, const vector<vector<vector<T>>>& VVV) {if (VVV.empty()) return os << "[[[]]]";os << "["<< "\n";int cnt = 0;for (auto&& VV : VVV) {os << cnt++ << VV << "\n\n";}os << "]";return os;}template <class T> ostream& operator<<(ostream& os, const set<T>& SS) {if (SS.empty()) return os << "[]";os << "[";auto ii = SS.begin();for (; ii != SS.end(); ii++) os << *ii << (ii == prev(SS.end()) ? "]" : ",");return os;}template <class Key, class Tp> ostream& operator<<(ostream& os, const map<Key, Tp>& MM) {if (MM.empty()) return os << "[{:}]";os << "[";auto ii = MM.begin();for (; ii != MM.end(); ii++)os << "{" << ii->first << ":" << ii->second << "}" << (ii == prev(MM.end()) ? "]" : ",");return os;}void debug_out() { cout << endl; }void debug_out_vl(vector<ll> V) {const int MAX_SIZE = min((int)V.size(), MAX_DUBUG_SIZE);cout << "\033[33m";if (V.empty()) {cout << "[]" << endl;return;}cout << "[";for (int i = 0; i < MAX_SIZE; i++) {if (V[i] == LINF)cout << "INF";elsecout << V[i];if (i == (int)V.size() - 1)cout << "]\n";else if (i == MAX_DUBUG_SIZE - 1)cout << ",...\n";elsecout << ",";}return;}void debug_out_vvl(vector<vector<ll>> VV) {cout << "\033[33m";if (VV.empty()) {cout << "[[]]" << endl;return;}cout << "[\n";int MAX_ROW = min((int)VV.size(), MAX_DUBUG_SIZE);for (int i = 0; i < MAX_ROW; i++) {const int MAX_COLUMN = min((int)VV[i].size(), MAX_DUBUG_SIZE);if (VV[i].empty()) {cout << "[]" << endl;continue;}cout << "[";for (int j = 0; j < MAX_COLUMN; j++) {if (VV[i][j] == LINF)cout << "INF";elsecout << VV[i][j];if (j == (int)VV[i].size() - 1)cout << "]\n";else if (j == MAX_DUBUG_SIZE - 1)cout << ",...\n";elsecout << ",";}if (i != (int)VV.size() - 1 and i == MAX_DUBUG_SIZE - 1) {cout << ":\n:\033[m\n";return;}}cout << "]\033[m\n";return;}template <class Head, class... Tail> void debug_out(Head H, Tail... T) {if constexpr (std::is_same_v<Head, vector<ll>>) {debug_out_vl(H);}else if constexpr (std::is_same_v<Head, vector<vector<ll>>>) {debug_out_vvl(H);}else {cout << "\033[33m" << H << "\033[m ";}debug_out(T...);}template <class T> bool chmax(T& a, const T& b) {if (a < b) {a = b;return true;}return false;}template <class T> bool chmin(T& a, const T& b) {if (b < a) {a = b;return true;}return false;}template <class T> T pwr(T x, ll n) {T res = 1;for (int i = 0; i < n; i++) {res *= x;}return res;}ll fact(ll n) {ll res = 1;for (ll i = 1; i <= n; i++) res *= i;return res;}ll comb(ll n, ll r) { // comb(60, 30)までオーバーフローなしll res = 1;for (int i = 1; i <= r; i++) {res *= n--;res /= i;}return res;}ll ctoll(char c) {if (c < '0' or '9' < c) {cerr << "\n\033[33m ctoll に '0'~'9' 以外の文字が入りました.\033[m\n" << endl;}return ll(c - '0');}char lltoc(ll n) {if (n < 0 or 9 < n) {cerr << "\n\033[33m lltoc に 0 ~ 9 以外の数字が入りました.\033[m\n" << endl;}return char(n + '0');}