結果

問題 No.2260 Adic Sum
ユーザー nononnonon
提出日時 2023-07-13 18:11:02
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,853 ms / 2,000 ms
コード長 1,231 bytes
コンパイル時間 2,291 ms
コンパイル使用メモリ 203,500 KB
最終ジャッジ日時 2025-02-15 10:14:30
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
using db = long double;
using ch = char;
using bl = bool;
using st = string;
using pll = pair<ll,ll>;
using psl = pair<st,ll>;
using vst = vector<st>;
using vch = vector<ch>;
using vvch = vector<vch>;
using vbl = vector<bl>;
using vvbl = vector<vbl>;
using vdb = vector<db>;
using vpll = vector<pll>;
using vpsl = vector<psl>;
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;
using vvvvi = vector<vvvi>;
using vll = vector<ll>;
using vvll = vector<vll>;
using vvvll = vector<vvll>;
using vvvvll = vector<vvvll>;
using vvvvvll = vector<vvvvll>;
#define all(A) A.begin(),A.end()
#define rep(i, n) for (ll i = 0; i < (ll)(n); i++)
#define rrep(i,a,b) for(int i=(ll)(a);i<=(ll)(b);i++)

int main() {
    ll N,P;
    cin>>N>>P;
    ll p=P;
    vll list;
    while(p<1e9){
        list.push_back(p);
        p*=P;
    }
    ll n=list.size();
    vector<map<ll,ll>> A(n);
    rep(i,N){
        ll a;
        cin>>a;
        rep(j,n){
            A[j][a%list[j]]++;
        }
    }
    ll ans=0;
    rep(j,n){
        for(auto x:A[j]){
            ll y=x.second;
            ans+=y*(y-1)/2;
        }
    }
    cout << ans << endl;
    return 0;
}
0