結果
問題 | No.2260 Adic Sum |
ユーザー |
|
提出日時 | 2023-04-08 01:49:17 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 861 ms / 2,000 ms |
コード長 | 1,030 bytes |
コンパイル時間 | 3,002 ms |
コンパイル使用メモリ | 252,324 KB |
実行使用メモリ | 10,240 KB |
最終ジャッジ日時 | 2024-10-06 07:24:28 |
合計ジャッジ時間 | 9,146 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
#include <bits/stdc++.h>using namespace std;using ch = char;using ll = long long;using ld = long double;using db = double;using st = string;using vdb = vector<db>;using vvdb = vector<vdb>;using vl = vector<ll>;using vvl = vector<vl>;using vvvl = vector<vvl>;using vd = vector<ld>;using vvd = vector<vd>;using vs = vector<st>;using vvs = vector<vs>;using vc = vector<ch>;using vvc = vector<vc>;using vb = vector<bool>;using vvb = vector<vb>;using vvvb = vector<vvb>;const ll mod = 998244353;const ll MOD = 1000000007;const ll INF = 1000000000000000000LL;using pall = pair<ll,ll>;using vp = vector<pall>;int main(){ll N,P;cin>>N>>P;vl A(N);ll MAX=0;for(int i=0; i<N; i++){cin>>A[i];MAX = max(MAX,A[i]);}ll ans = 0;for(ll i=P; i<=MAX; i*=P){map<ll,ll> X;for(int j=0; j<N; j++){X[A[j]%i]++;}for(auto mp:X){ans += mp.second * (mp.second - 1) /2;}}cout << ans << endl;}