結果
問題 |
No.108 トリプルカードコンプ
|
ユーザー |
![]() |
提出日時 | 2024-08-28 02:52:50 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 135 ms / 5,000 ms |
コード長 | 597 bytes |
コンパイル時間 | 472 ms |
コンパイル使用メモリ | 82,372 KB |
実行使用メモリ | 89,312 KB |
最終ジャッジ日時 | 2024-08-28 02:52:54 |
合計ジャッジ時間 | 3,168 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
def list3(a, b, c, *, val=0): return [[[val] * c for _ in range(b)] for _ in range(a)] N = int(input()) A = list(map(int, input().split())) # 0枚がa種。1枚がb種、2枚がc種 def f(a, b, c): if a+b+c == 0: return 0 if memo[a][b][c] != 0: return memo[a][b][c] res = 1 if a > 0: res += f(a-1, b+1, c) * a / N if b > 0: res += f(a, b-1, c+1) * b / N if c > 0: res += f(a, b, c-1) * c / N res *= N / (a+b+c) memo[a][b][c] = res return res a = A.count(0) b = A.count(1) c = A.count(2) memo = list3(N+1, N+1, N+1, val=0) ans = f(a, b, c) print(ans)