結果
問題 |
No.108 トリプルカードコンプ
|
ユーザー |
|
提出日時 | 2023-07-06 00:05:58 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 529 ms / 5,000 ms |
コード長 | 567 bytes |
コンパイル時間 | 124 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 45,152 KB |
最終ジャッジ日時 | 2024-07-19 18:24:38 |
合計ジャッジ時間 | 2,872 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
import sys sys.setrecursionlimit(10**6) n = int(input()) A = list(map(int, input().split())) C = [0, 0, 0, 0] for i in range(n): a = min(A[i], 3) C[a] += 1 memo = dict() memo[(0, 0, 0, n)] = 0.0 def E(X): if X in memo: return memo[X] Z = list(X) res = 0.0 for i in range(3): if Z[i] == 0: continue Z[i] -= 1 Z[i + 1] += 1 res += (Z[i] + 1) * (E(tuple(Z)) + 1) Z[i] += 1 Z[i + 1] -= 1 res += Z[3] res /= n - Z[3] memo[X] = res return res print(E(tuple(C)))