結果

問題 No.1688 Veterinarian
ユーザー tktk_snsntktk_snsn
提出日時 2021-09-24 22:44:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 777 ms / 3,000 ms
コード長 1,113 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 82,008 KB
実行使用メモリ 159,088 KB
最終ジャッジ日時 2024-07-05 11:03:56
合計ジャッジ時間 7,651 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 106 ms
78,948 KB
testcase_01 AC 777 ms
159,088 KB
testcase_02 AC 297 ms
100,564 KB
testcase_03 AC 104 ms
78,576 KB
testcase_04 AC 104 ms
79,096 KB
testcase_05 AC 105 ms
79,016 KB
testcase_06 AC 770 ms
159,080 KB
testcase_07 AC 108 ms
78,832 KB
testcase_08 AC 770 ms
159,068 KB
testcase_09 AC 754 ms
157,420 KB
testcase_10 AC 560 ms
132,468 KB
testcase_11 AC 430 ms
116,648 KB
testcase_12 AC 151 ms
82,644 KB
testcase_13 AC 479 ms
120,944 KB
testcase_14 AC 225 ms
91,228 KB
testcase_15 AC 713 ms
153,068 KB
testcase_16 AC 240 ms
92,632 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

U = 50
A, B, C, N = map(int, input().split())

# dp[i,j,k]: (i,j,k)枚もってる確率
dp = [[[0] * (U + 1) for _ in range(U + 1)] for _ in range(U + 1)]
dp[A][B][C] = 1.0

for _ in range(N):
    ndp = [[[0] * (U + 1) for _ in range(U + 1)] for _ in range(U + 1)]
    for a in range(U+1):
        for b in range(U+1):
            for c in range(U+1):
                tot = a + b + c
                tot = tot * (tot - 1) // 2
                if tot == 0:
                    continue

                pa = a * (a - 1) // 2 / tot
                ndp[a-1][b][c] += dp[a][b][c] * pa

                pb = b * (b - 1) // 2 / tot
                ndp[a][b-1][c] += dp[a][b][c] * pb

                pc = c * (c - 1) // 2 / tot
                ndp[a][b][c-1] += dp[a][b][c] * pc

                ndp[a][b][c] += dp[a][b][c] * (1 - pa - pb - pc)
    dp = ndp

x, y, z = 0, 0, 0
for a in range(U+1):
    for b in range(U+1):
        for c in range(U+1):
            x += (A - a) * dp[a][b][c]
            y += (B - b) * dp[a][b][c]
            z += (C - c) * dp[a][b][c]

print("{:.12f} {:.12f} {:.12f}".format(x, y, z))
0