結果

問題 No.2260 Adic Sum
ユーザー lddlinan
提出日時 2023-04-07 22:03:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 899 bytes
コンパイル時間 707 ms
コンパイル使用メモリ 92,696 KB
最終ジャッジ日時 2025-02-12 01:22:01
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 3
other WA * 33
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:31:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   31 |     scanf("%d %d", &n, &p);
      |     ~~~~~^~~~~~~~~~~~~~~~~
main.cpp:32:30: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   32 |     for (i=0; i<n; i++) scanf("%d", &as[i]);
      |                         ~~~~~^~~~~~~~~~~~~~
main.cpp:40:14: warning: ‘r’ may be used uninitialized [-Wmaybe-uninitialized]
   40 |             r+=t;
      |             ~^~~
main.cpp:30:11: note: ‘r’ was declared here
   30 |     LL b, r, t;
      |           ^

ソースコード

diff # 

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include <map>
#include <vector>
#include <queue>
#include <deque>
#include <set>
#include <stack>
#include <algorithm>
#include <array>
#include <unordered_set>
#include <unordered_map>
#include <string>
using namespace std;

bool rcmp(int a, int b) { return a>b; }
typedef long long LL;
class mypcmp {
public:
    bool operator()(const int& a, const int& b) {
        return a<b;
    }
};


int as[100004];
int main() {
    int n, i, p, mv, c;
    LL b, r, t;
    scanf("%d %d", &n, &p);
    for (i=0; i<n; i++) scanf("%d", &as[i]);
    mv=as[0]; for (i=1; i<n; i++) mv=max(mv, as[i]);
    for (b=p; b<=mv; b*=p) {
        map<int, int> cs;
        for (i=0; i<n; i++) cs[as[i]%b]++;
        for (auto x: cs) {
            c=x.second;
            t=c; t*=c-1; t/=2;
            r+=t;
        }
    }
    printf("%lld\n", r);
    return 0;
}
0