結果

問題 No.2709 1975 Powers
ユーザー 259-Momone259-Momone
提出日時 2024-11-01 20:40:38
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 589 ms / 2,000 ms
コード長 826 bytes
コンパイル時間 4,166 ms
コンパイル使用メモリ 339,128 KB
実行使用メモリ 12,420 KB
最終ジャッジ日時 2024-11-01 20:40:48
合計ジャッジ時間 9,515 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <atcoder/math>
#include <bits/extc++.h>

int main() {
    using namespace std;
    unsigned N, P, Q;
    cin >> N >> P >> Q;
    vector<unsigned> A(N);
    for (auto &a : A)
        cin >> a;
    ranges::sort(A);

    unordered_map<unsigned, unsigned long> dp_0, dp_1, dp_2, dp_3;
    for (const auto &a : A) {
        const auto v5{atcoder::pow_mod(5, a, P)};
        for (const auto &[k, v] : dp_2)
            dp_3[(k + v5) % P] += v;
        const auto v7{atcoder::pow_mod(7, a, P)};
        for (const auto &[k, v] : dp_1)
            dp_2[(k + v7) % P] += v;
        const auto v9{atcoder::pow_mod(9, a, P)};
        for (const auto &[k, v] : dp_0)
            dp_1[(k + v9) % P] += v;
        const auto v10{atcoder::pow_mod(10, a, P)};
        dp_0[v10]++;
    }

    cout << dp_3[Q] << endl;
    return 0;
}
0