結果

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

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
8,260 KB
testcase_01 AC 3 ms
8,260 KB
testcase_02 AC 39 ms
49,228 KB
testcase_03 AC 561 ms
135,244 KB
testcase_04 AC 900 ms
153,676 KB
testcase_05 AC 644 ms
139,340 KB
testcase_06 AC 288 ms
104,524 KB
testcase_07 AC 7 ms
18,504 KB
testcase_08 AC 272 ms
104,524 KB
testcase_09 AC 681 ms
139,340 KB
testcase_10 AC 7 ms
18,504 KB
testcase_11 AC 58 ms
59,468 KB
testcase_12 AC 109 ms
75,852 KB
testcase_13 AC 699 ms
141,388 KB
testcase_14 AC 168 ms
88,140 KB
testcase_15 AC 191 ms
90,184 KB
testcase_16 AC 564 ms
133,196 KB
testcase_17 AC 17 ms
32,840 KB
testcase_18 AC 136 ms
79,948 KB
testcase_19 AC 879 ms
153,676 KB
testcase_20 AC 31 ms
45,132 KB
testcase_21 AC 273 ms
102,476 KB
testcase_22 AC 922 ms
159,820 KB
testcase_23 AC 994 ms
159,820 KB
testcase_24 AC 982 ms
159,820 KB
testcase_25 AC 988 ms
159,820 KB
testcase_26 AC 1,002 ms
159,820 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