結果

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

ソースコード

diff #

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

0