結果

問題 No.2260 Adic Sum
コンテスト
ユーザー akinyan
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#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;
}
0