結果

問題 No.108 トリプルカードコンプ
コンテスト
ユーザー bamboo ryo
提出日時 2020-12-26 22:17:39
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,766 ms / 5,000 ms
コード長 644 bytes
コンパイル時間 198 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 52,864 KB
最終ジャッジ日時 2024-09-25 02:24:54
合計ジャッジ時間 19,088 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

n = int(input())

a = [0, 0, 0]
for v in map(int, input().split()):
    if v <= 2:
        a[v] += 1

dp = [[[0 for _ in range(n+3)] for _ in range(n+3)] for _ in range(n+3)]

for i in range(n+1):
    for j in range(n+1):
        for k in range(n+1):
            nonzero = i + j + k
            if nonzero == 0:
                dp[i][j][k] = 0
                continue

            dp[i][j][k] += n/nonzero
            if (i-1 >= 0): dp[i][j][k] += dp[i-1][j+1][k] * i/nonzero
            if (j-1 >= 0): dp[i][j][k] += dp[i][j-1][k+1] * j/nonzero
            if (k-1 >= 0): dp[i][j][k] += dp[i][j][k-1] * k/nonzero

print(dp[a[0]][a[1]][a[2]])
0