結果
問題 | No.2260 Adic Sum |
ユーザー |
![]() |
提出日時 | 2023-04-07 21:23:26 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 201 ms / 2,000 ms |
コード長 | 1,041 bytes |
コンパイル時間 | 1,924 ms |
コンパイル使用メモリ | 200,532 KB |
最終ジャッジ日時 | 2025-02-12 00:13:21 |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 33 |
ソースコード
#include <bits/stdc++.h>using namespace std;typedef long long ll;template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return true; } return false; }template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return true; } return false; }#define all(x) (x).begin(),(x).end()#define fi first#define se second#define mp make_pair#define si(x) int(x.size())const int mod=998244353,MAX=300005,INF=1<<30;int main(){std::ifstream in("text.txt");std::cin.rdbuf(in.rdbuf());cin.tie(0);ios::sync_with_stdio(false);ll N,P;cin>>N>>P;vector<ll> A(N);ll ans=0;for(int i=0;i<N;i++) cin>>A[i];ll wa=1;while(1){wa*=P;if(wa>1000000000) break;vector<ll> B(N);for(int i=0;i<N;i++){B[i]=A[i]%wa;}sort(all(B));int i=0;while(i<N){int j=i;while(j<N&&B[i]==B[j]) j++;ans+=(ll)(j-i)*(j-i-1)/2;i=j;}}cout<<ans<<endl;}