#1023916 (C++23) No.2709 1975 Powers

• 問題: No.2709 1975 Powers
• ユーザー: commy
• 提出日時: 2024-11-01 20:34:50
• 言語: C++23 (gcc 13.3.0 + boost 1.87.0)
• 結果: AC
• 実行時間: 133 ms / 2,000 ms
• コード長: 2,895 bytes
• コンパイル時間: 1,549 ms
• コンパイル使用メモリ: 114,504 KB
• 実行使用メモリ: 75,136 KB
• 最終ジャッジ日時: 2024-11-01 20:34:55
• 合計ジャッジ時間: 4,186 ms
• ジャッジサーバーID （参考情報）: judge1 / judge3

ファイルパターン	結果
sample	AC * 2
other	AC * 25

ソースコード

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
#include <algorithm>
#include <iostream>
#include <numeric>
#include <string>
#include <tuple>
#include <utility>
#include <vector>
#include <limits>
#define rep(i, a, b) for (int i = int(a); i < int(b); i++)
using namespace std;
using ll = long long int;  // NOLINT
using P = pair<ll, ll>;
// clang-format off
#ifdef _DEBUG_
#define dump(...) do{ cerr << __LINE__ << ":\t" << #__VA_ARGS__ << " = "; debug_print(__VA_ARGS__); } while(false)
template<typename T, typename... Ts> void debug_print(const T &t, const Ts &...ts) { cerr << t; ((cerr << ", " << ts), ...); cerr << endl; }
#else
#define dump(...) do{ } while(false)
#endif
template<typename T> vector<T> make_v(size_t a, T b) { return vector<T>(a, b); }
template<typename... Ts> auto make_v(size_t a, Ts... ts) { return vector<decltype(make_v(ts...))>(a, make_v(ts...)); }
template<typename T> bool chmin(T &a, const T& b) { if (a > b) {a = b; return true; } return false; }
template<typename T> bool chmax(T &a, const T& b) { if (a < b) {a = b; return true; } return false; }
template<typename T, typename... Ts> void print(const T& t, const Ts&... ts) { cout << t; ((cout << ' ' << ts), ...); cout << '\n'; }
constexpr static struct PositiveInfinity { template<typename T> constexpr operator T() const { return numeric_limits<T>::max() / 2; } constexpr auto 
    operator-() const; } inf;  // NOLINT
constexpr static struct NegativeInfinity { template<typename T> constexpr operator T() const { return numeric_limits<T>::lowest() / 2; } constexpr 
    auto operator-() const; } NegativeInfinityVal;
constexpr auto PositiveInfinity::operator-() const { return NegativeInfinityVal; }
constexpr auto NegativeInfinity::operator-() const { return inf; }
// clang-format on
ll modpow(ll a, ll n, ll mod) {
    if (n == 0)
        return 1;
    if (n % 2 == 0) {
        ll res = modpow(a, n / 2, mod);
        return (res * res) % mod;
    }
    return (a * modpow(a, n - 1, mod)) % mod;
}
int main() {
    cin.tie(nullptr);
    ios::sync_with_stdio(false);
    int n, p, q;
    cin >> n >> p >> q;
    vector<ll> a(n);
    rep(i, 0, n) cin >> a[i];
    sort(a.begin(), a.end());
    vector<ll> cnt7(n, 0), cnt9(n, 0), cnt10(n, 0);
    rep(i, 0, n) {
        cnt7[i] = modpow(7, a[i], p);
        cnt9[i] = modpow(9, a[i], p);
        cnt10[i] = modpow(10, a[i], p);
    }
    auto cnt5 = make_v(n + 1, p + 1, 0);
    for (int i = n - 1; i >= 0; i--) {
        ll v = modpow(5, a[i], p);
        cnt5[i] = cnt5[i + 1];
        cnt5[i][v]++;
    }
    ll ans = 0;
    rep(i, 0, n) {
        rep(j, i + 1, n) {
            rep(k, j + 1, n) {
                ll tmp = (cnt10[i] + cnt9[j] + cnt7[k]) % p;
                ll r = (q - tmp + p) % p;
                if (cnt5[k + 1][r] > 0) {
                    dump(i, j, k, r);
                }
                ans += cnt5[k + 1][r];
            }
        }
    }
    print(ans);
    return 0;
}
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX