結果

問題 No.2709 1975 Powers
ユーザー rurunrurun
提出日時 2024-03-31 13:59:40
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,738 ms / 2,000 ms
コード長 1,116 bytes
コンパイル時間 1,985 ms
コンパイル使用メモリ 210,292 KB
実行使用メモリ 287,024 KB
最終ジャッジ日時 2024-09-30 21:47:18
合計ジャッジ時間 22,892 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1 ms
5,248 KB
testcase_02 AC 106 ms
86,520 KB
testcase_03 AC 972 ms
142,824 KB
testcase_04 AC 1,545 ms
242,236 KB
testcase_05 AC 1,148 ms
193,204 KB
testcase_06 AC 447 ms
62,968 KB
testcase_07 AC 22 ms
28,232 KB
testcase_08 AC 533 ms
181,324 KB
testcase_09 AC 1,167 ms
213,112 KB
testcase_10 AC 23 ms
30,028 KB
testcase_11 AC 88 ms
42,872 KB
testcase_12 AC 253 ms
145,704 KB
testcase_13 AC 1,224 ms
260,388 KB
testcase_14 AC 279 ms
80,812 KB
testcase_15 AC 397 ms
173,264 KB
testcase_16 AC 925 ms
111,160 KB
testcase_17 AC 35 ms
33,348 KB
testcase_18 AC 298 ms
151,872 KB
testcase_19 AC 1,500 ms
216,012 KB
testcase_20 AC 90 ms
80,156 KB
testcase_21 AC 545 ms
179,864 KB
testcase_22 AC 1,704 ms
287,024 KB
testcase_23 AC 1,728 ms
228,352 KB
testcase_24 AC 1,671 ms
134,360 KB
testcase_25 AC 1,735 ms
220,364 KB
testcase_26 AC 1,738 ms
246,960 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using lint = long long;

template<typename T>
T modpow(T a, T b, T mod=998244353) {
  T res = 1;
  while (b > 0) {
    if (b&1) (res *= a) %= mod;
    (a *= a) %= mod;
    b >>= 1;
  }
  return res;
}

int main() {
  lint n, p, q;
  cin >> n >> p >> q;
  vector<lint> vec(n);
  for (int i = 0; i < n; i++) cin >> vec[i];
  sort(vec.begin(), vec.end());
  vector<vector<lint>> cnt(n+1, vector<lint>(p));  // iより後ろでjあまるものの数
  vector rui = cnt;


  for (int i = 0; i < n; i++) {
    cnt[i][modpow(5LL, vec[i], p)]++;
  }
  for (int i = n-1; i >= 0; i--) {
    for (int j = 0; j < p; j++) {
      rui[i][j] = rui[i+1][j]+cnt[i+1][j];
    }
  }
  lint ans = 0;
  for (int i = 0; i < n; i++) {
    for (int j = i+1; j < n; j++) {
      for (int k = j+1; k < n; k++) {
        lint now = modpow(10LL, vec[i], p)+modpow(9LL, vec[j], p)+ modpow(7LL, vec[k], p);
        now %= p;
        ans += rui[k][(q+p-now)%p];
      }
    }
  }
  cout << ans << endl;
  return 0;
  for (auto x : rui) {
    for (auto y : x) cout << y << " ";
    cout << endl;
  }
}
0