結果

問題 No.1688 Veterinarian
ユーザー tktk_snsntktk_snsn
提出日時 2021-09-24 22:44:24
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 861 ms / 3,000 ms
コード長 1,113 bytes
コンパイル時間 309 ms
コンパイル使用メモリ 86,936 KB
実行使用メモリ 161,160 KB
最終ジャッジ日時 2023-09-18 21:52:47
合計ジャッジ時間 9,311 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge11
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 153 ms
82,004 KB
testcase_01 AC 855 ms
161,124 KB
testcase_02 AC 368 ms
101,980 KB
testcase_03 AC 149 ms
82,232 KB
testcase_04 AC 149 ms
81,728 KB
testcase_05 AC 151 ms
81,812 KB
testcase_06 AC 861 ms
161,160 KB
testcase_07 AC 152 ms
82,008 KB
testcase_08 AC 855 ms
161,036 KB
testcase_09 AC 841 ms
160,936 KB
testcase_10 AC 629 ms
135,572 KB
testcase_11 AC 496 ms
118,872 KB
testcase_12 AC 198 ms
85,520 KB
testcase_13 AC 546 ms
124,504 KB
testcase_14 AC 287 ms
93,788 KB
testcase_15 AC 815 ms
155,388 KB
testcase_16 AC 294 ms
95,280 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