結果
問題 |
No.108 トリプルカードコンプ
|
ユーザー |
![]() |
提出日時 | 2023-10-30 17:11:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 100 ms / 5,000 ms |
コード長 | 1,368 bytes |
コンパイル時間 | 456 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 85,120 KB |
最終ジャッジ日時 | 2024-09-25 17:19:29 |
合計ジャッジ時間 | 2,923 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 20 |
ソースコード
N = int(input()) A = list(map(int, input().split())) dp = [[[-1] * (N + 1) for _ in range(N + 1)] for _ in range(N + 1)] dp[N][N][N] = 0 for k in range(N, -1, -1): for j in range(N, k - 1, -1): for i in range(N, j - 1, -1): if k == N: continue elif j == N: dp[i][j][k] = (1 + dp[i][j][k + 1] * (j - k) / N) * N / (N - k) elif j == N: dp[i][j][k] = ( (dp[i + 1][j][k] * (N - i) / N + dp[i][j][k + 1] * (j - k) / N) * N / (N - k) ) elif i == N: dp[i][j][k] = ( (1 + dp[i][j + 1][k] * (i - j) / N + dp[i][j][k + 1] * (j - k) / N) * N / (N - k) ) else: dp[i][j][k] = ( ( 1 + dp[i + 1][j][k] * (N - i) / N + dp[i][j + 1][k] * (i - j) / N + dp[i][j][k + 1] * (j - k) / N ) * N / (N - k) ) i, j, k = 0, 0, 0 for a in A: if a >= 3: i += 1 j += 1 k += 1 elif a >= 2: i += 1 j += 1 elif a >= 1: i += 1 print(dp[i][j][k])