結果

問題 No.2260 Adic Sum
ユーザー hiro71687k
提出日時 2023-04-07 21:32:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 988 ms / 2,000 ms
コード長 693 bytes
コンパイル時間 4,218 ms
コンパイル使用メモリ 255,680 KB
最終ジャッジ日時 2025-02-12 00:37:19
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
using ll=long long;
using ld=long double;
ld pie=3.141592653589793;
ll inf=144494;
ll mod=1000000007;
int main(){
    ll n,p;
    cin >> n >> p;
    vector<ll>a(n);
    for (ll i = 0; i < n; i++)
    {
        cin >> a[i];
    }
    ll now=1;
    ll ans=0;
    for (ll i = 1; i >=0; i++)
    {
        now*=p;
        if (now>1000000000)
        {
            break;
        }
        map<ll,ll>memo;
        for (ll j = 0; j < n; j++)
        {
            memo[a[j]%now]+=1;
        }
        for(auto v:memo){
            ans+=(v.second*(v.second-1))/2;
        }
    }
    cout << ans << endl;
}
0