結果
問題 | No.2260 Adic Sum |
ユーザー |
👑 ![]() |
提出日時 | 2023-04-07 22:28:37 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 185 ms / 2,000 ms |
コード長 | 1,028 bytes |
コンパイル時間 | 897 ms |
コンパイル使用メモリ | 80,800 KB |
最終ジャッジ日時 | 2025-02-12 01:50:33 |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
#include <iostream> #include <string> #include <vector> #include <algorithm> #include <atcoder/modint> using namespace std; using i32 = int; using u32 = unsigned int; using i64 = long long; using u64 = unsigned long long; #define rep(i,n) for(int i=0; i<(int)(n); i++) const i64 INF = 1001001001001001001; using Modint = atcoder::static_modint<998244353>; int main(){ int N, P; cin >> N >> P; vector<int> A(N); rep(i,N) cin >> A[i]; i64 ans = 0; for(int p=P; ; p*=P){ vector<int> B(N); rep(i,N) B[i] = A[i] % p; sort(B.begin(), B.end()); int cnt = 1; for(int i=1; i<N; i++){ if(B[i] == B[i-1]){ ans += cnt; cnt++; } else { cnt = 1; } } if((i64)p * P > 1'000'000'000) break; } cout << ans << endl; return 0; } struct ios_do_not_sync{ ios_do_not_sync(){ ios::sync_with_stdio(false); cin.tie(nullptr); } } ios_do_not_sync_instance;