結果

問題 No.2260 Adic Sum
コンテスト
ユーザー nagisa5101
提出日時 2023-04-07 21:40:42
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,954 bytes
コンパイル時間 16,952 ms
コンパイル使用メモリ 330,828 KB
最終ジャッジ日時 2025-02-12 00:52:15
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>

#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")

using namespace std;
using namespace atcoder;

#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define repll(i, n) for (long long i = 0; i < (long long)(n); i++)
#define rep2(i, n, m) for (int i = n; i < (int)(m); i++)
#define repll2(i, n, m) for (long long i = n; i < (long long)(m); i++)
#define all(v) v.begin(),v.end()
using ll=long long;
using ld=long double;
using vi=vector<int>;
using vvi=vector<vi>;
using vvvi=vector<vvi>;
using vl=vector<ll>;
using vvl=vector<vl>;
using vvvl=vector<vvl>;
using vld=vector<ld>;
using vvld=vector<vld>;

int dx[8]={1,0,-1,0,1,1,-1,-1};
int dy[8]={0,1,0,-1,1,-1,1,-1};

const double PI = acos(-1);
//const ll MOD=1e9+7;
//const ll MOD=998244353;
const ll INF=(1LL<<60);
const int INF2=(1<<30);
//using mint=modint1000000007;
//using mint=modint998244353;

pair<ll,ll> calc(ll n,ll p){
    //nは何回pで割り切れるか？
    //最終的なmodも求める
    ll c=0;
    while((n%p)==0){
        c++;
        n/=p;
    }
    return {c,n};
}

int main() {
    ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    ll n,p;cin>>n>>p;
    vl cnt(30,0);
    ll M=0;
    ll now=1;
    while(now<1e9){
        now*=p;
        M++;
    }
    vector<vector<map<ll,ll>>> cnt_mod(30,vector<map<ll,ll>>(M));

    rep(i,n){
        ll a;cin>>a;
        auto pp=calc(a,p);
        ll k=pp.first,mod=pp.second;
        //cout<<k<<" "<<mod<<endl;
        now=1;
        rep(j,M){
            now*=p;
            cnt_mod[k][j][mod%now]++;
        }
        cnt[k]++;
    }
    ll ans=0;
    repll(i,30){
        rep2(j,i+1,30)ans+=cnt[i]*cnt[j]*i;
        ans+=cnt[i]*(cnt[i]-1)*i/2;
        rep(j,M){
            for(auto p:cnt_mod[i][j]){
                ll c=p.second;
                ans+=c*(c-1)/2;
            }
        }
    }
    cout<<ans<<endl;
    return 0;
}
0