結果

問題 No.2709 1975 Powers
コンテスト
ユーザー tobbie
提出日時 2024-04-27 16:01:25
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,015 bytes
コンパイル時間 2,116 ms
コンパイル使用メモリ 178,380 KB
実行使用メモリ 17,096 KB
最終ジャッジ日時 2024-11-15 19:32:28
合計ジャッジ時間 77,995 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other TLE * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define rep(i, j, n) for (int i = (int)j; i < (int)(n); i++)
using ll = long long int;
using vi = vector<int>;
using vvi = vector<vi>;
using vvvi = vector<vvi>;

int main() {
  int N, P, Q;
  cin >> N >> P >> Q;
  vi A(N);
  rep(i, 0, N)
    cin >> A[i];
  sort(A.begin(), A.end());
  vi o = {10, 9, 7, 5};
  vvi a(N, vector<int> (4));
  rep(i, 0, N) {
    rep(j, 0, i+1) {
      int p = 1;
      rep(k, 0, A[i])
	p = ((ll)p * (ll)o[j]) % P;
      a[i][j] = p;
    }
  }
  vvvi y(P, vvi (N, vi (4, -1)));
  auto dfs = [&](auto dfs, int x, int n, int depth, int ans) {
    x = (x + a[n][depth]) % P;
    y[x][n][depth] = x;
    if (depth == 3) {
      if (x % P == Q)
	return ans + 1;
      else
	return ans;
    }
    rep(i, n+1, N) {
      if (y[x][i][depth+1] >= 0)
	ans = y[x][i][depth+1];
      else
	ans = dfs(dfs, x, i, depth+1, ans);
    }
    return ans;
  };
  int ans = 0;
  rep(i, 0, N)
    ans += dfs(dfs, 0, i, 0, 0);
  cout << ans << endl;
  return 0;
}
0