結果
問題 | No.2260 Adic Sum |
ユーザー |
|
提出日時 | 2023-04-07 21:59:01 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 109 ms / 2,000 ms |
コード長 | 1,137 bytes |
コンパイル時間 | 2,197 ms |
コンパイル使用メモリ | 203,844 KB |
最終ジャッジ日時 | 2025-02-12 01:19:17 |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
// 提出時にassertはオフ #ifndef DEBUG #ifndef NDEBUG #define NDEBUG #endif #endif #include <bits/stdc++.h> using namespace std; using ll = long long; #define ALL(x) (x).begin(), (x).end() template <class T> using vec = vector<T>; int N, P; ll ans = 0; // 同じmodPの集合に対し、そのmodを引いて、Pで割る void groupByModP(const vec<int> &A) { if(A.size() <= 1) return; int curMod = A[0] % P; unordered_map<int, vec<int>> newGroup; for(int num : A) { int newNum = num + P - curMod; assert(newNum % P == 0); newNum /= P; newGroup[newNum % P].push_back(newNum); } ans += (ll)A.size() * (A.size() - 1) / 2; for(const pair<int, vec<int>> &p : newGroup) { groupByModP(p.second); } } int main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cin >> N >> P; vec<int> A(N); for(int &a : A) cin >> a; unordered_map<int, vec<int>> group; for(int num : A) { group[num % P].push_back(num); } for(const pair<int, vec<int>> &p : group) { groupByModP(p.second); } cout << ans << "\n"; }