結果

問題 No.2709 1975 Powers
ユーザー くけくけ
提出日時 2024-03-31 15:18:23
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 992 ms / 2,000 ms
コード長 1,918 bytes
コンパイル時間 4,556 ms
コンパイル使用メモリ 265,472 KB
実行使用メモリ 159,056 KB
最終ジャッジ日時 2024-09-30 20:32:57
合計ジャッジ時間 16,828 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
7,876 KB
testcase_01 AC 4 ms
7,504 KB
testcase_02 AC 37 ms
48,716 KB
testcase_03 AC 558 ms
135,248 KB
testcase_04 AC 899 ms
152,528 KB
testcase_05 AC 639 ms
138,956 KB
testcase_06 AC 287 ms
105,684 KB
testcase_07 AC 8 ms
17,232 KB
testcase_08 AC 271 ms
103,248 KB
testcase_09 AC 672 ms
139,604 KB
testcase_10 AC 8 ms
18,248 KB
testcase_11 AC 59 ms
58,320 KB
testcase_12 AC 108 ms
73,808 KB
testcase_13 AC 689 ms
141,008 KB
testcase_14 AC 167 ms
86,740 KB
testcase_15 AC 190 ms
90,320 KB
testcase_16 AC 563 ms
131,412 KB
testcase_17 AC 17 ms
31,188 KB
testcase_18 AC 134 ms
79,688 KB
testcase_19 AC 875 ms
151,896 KB
testcase_20 AC 32 ms
43,992 KB
testcase_21 AC 269 ms
101,836 KB
testcase_22 AC 948 ms
159,052 KB
testcase_23 AC 992 ms
159,056 KB
testcase_24 AC 976 ms
158,936 KB
testcase_25 AC 977 ms
158,924 KB
testcase_26 AC 975 ms
158,928 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long ll;
#define rep(i, n) for (int i = 0; i < n; i++)
#define chmin(x, y) x = min(x, y)
#define chmax(x, y) x = max(x, y)
const int MAX = 1e7 ;
int MOD = 998244353 ;
 
long long fac[MAX], finv[MAX], inv[MAX];
ll D[205][100005] ;

// テーブルを作る前処理
void COMinit() {
    fac[0] = fac[1] = 1;
    finv[0] = finv[1] = 1;
    inv[1] = 1;
    for (int i = 2; i < MAX; i++){
        fac[i] = fac[i - 1] * i % MOD;
        inv[i] = MOD - inv[MOD%i] * (MOD / i) % MOD;
        finv[i] = finv[i - 1] * inv[i] % MOD;
    }
}
 
// 二項係数計算 nCk
long long COM(int n, int k){
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD;
}
// 順列計算 nPk = n! / ( ( n - k ) ! )
long long PER(int n , int k){
    if (n < k) return 0;
    if (n < 0 || k < 0) return 0;
    return ( fac[n] * finv[n-k] ) % MOD ;
}
// べき乗計算
ll POW(ll a, ll n ) {
  if (n == 0) return 1;
  if (n == 1) return a % MOD;
  if (n % 2 == 1) return (a * POW(a, n - 1 )) % MOD;
  ll t = POW(a, n / 2 );
  return (t * t) % MOD ;
}


int main(){
   ll n , p , q ;
   cin >> n >> p >> q ;
   MOD = p ;
   vector<ll> A(n) ;
   rep(i,n) cin >> A[i] ;
   sort(A.begin(),A.end()) ;
   //reverse(A.begin(),A.end()) ;
   rep(j,n){
      D[j][POW(5,A[j])]++; 
   }
   for( ll j = n - 1 ; j > 0 ; j-- ){
      rep(i,100005){
         D[j-1][i] += D[j][i] ;
      }
   }
   ll ans = 0 ;
   for( ll a = 0 ; a < n ; a++ ){
      ll tmp1 = POW(10,A[a]) ;
      for( ll b = a + 1 ; b < n ; b++ ){
         ll tmp2 = POW(9,A[b]) ;
         for( ll c = b + 1 ; c < n ; c++ ){
            ll tmp = tmp1 + tmp2 +POW(7,A[c]) ;
            tmp %= MOD ;
            tmp =  ( q - tmp + MOD ) % MOD ;
            ans += D[c+1][tmp] ;
         }
      }
   }
   cout << ans << endl ;
}
0