結果
問題 |
No.108 トリプルカードコンプ
|
ユーザー |
|
提出日時 | 2021-01-03 20:19:05 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 120 ms / 5,000 ms |
コード長 | 925 bytes |
コンパイル時間 | 705 ms |
コンパイル使用メモリ | 82,060 KB |
実行使用メモリ | 85,748 KB |
最終ジャッジ日時 | 2024-10-13 14:28:02 |
合計ジャッジ時間 | 2,663 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
# トリプルカードコンプ N = int(input()) A = list(map(int,input().split())) # dp[i][j][k] = 0枚持ってるカードがi種類、1枚のカードがj種類、2枚のカードがk種類ある # 2枚持ってる状態からそのカードを当てると、管理対象から外れる cnt = [0] * 11 for a in A: cnt[a] += 1 dp = [[[0] * (N + 2) for j in range(N + 2)] for i in range(N + 2)] dp[0][0][0] = 0 for i in range(N + 1): for j in range(N + 1): for k in range(N + 1): has = i + j + k if has == 0: continue dp[i][j][k] = N / has # いずれかの対象カードを引くための試行回数の期待値 if k - 1 >= 0: dp[i][j][k] += dp[i][j][k - 1] * (k / has) if j - 1 >= 0: dp[i][j][k] += dp[i][j - 1][k + 1] * (j / has) if i - 1 >= 0: dp[i][j][k] += dp[i - 1][j + 1][k] * (i / has) print(dp[cnt[0]][cnt[1]][cnt[2]])