結果

問題 No.1688 Veterinarian
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2021-09-24 22:12:52
言語 PyPy3
(7.3.13)
結果
AC  
実行時間 264 ms / 3,000 ms
コード長 1,140 bytes
コンパイル時間 717 ms
コンパイル使用メモリ 87,328 KB
実行使用メモリ 91,800 KB
最終ジャッジ日時 2023-09-18 21:31:33
合計ジャッジ時間 3,665 ms
ジャッジサーバーID
(参考情報)
judge13 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 75 ms
70,976 KB
testcase_01 AC 77 ms
71,204 KB
testcase_02 AC 76 ms
71,528 KB
testcase_03 AC 76 ms
71,296 KB
testcase_04 AC 75 ms
71,212 KB
testcase_05 AC 75 ms
71,424 KB
testcase_06 AC 87 ms
76,440 KB
testcase_07 AC 95 ms
76,976 KB
testcase_08 AC 264 ms
91,800 KB
testcase_09 AC 254 ms
91,320 KB
testcase_10 AC 144 ms
78,348 KB
testcase_11 AC 150 ms
79,700 KB
testcase_12 AC 90 ms
76,576 KB
testcase_13 AC 114 ms
77,384 KB
testcase_14 AC 103 ms
77,124 KB
testcase_15 AC 156 ms
78,864 KB
testcase_16 AC 92 ms
76,388 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

A,B,C,n = map(int,input().split())
dp = [[[0]*(C+1) for i in range(B+1)] for j in range(A+1)]
dp[A][B][C] = 1
for i in range(n):
    ndp = [[[0]*(C+1) for i in range(B+1)] for j in range(A+1)]
    for a in range(A+1):
        for b in range(B+1):
            for c in range(C+1):
                if dp[a][b][c] == 0:
                    continue
                prob = dp[a][b][c]

                if a > 1:
                    ndp[a-1][b][c] += prob*a*(a-1)/(a+b+c)/(a+b+c-1)
                    ndp[a][b][c] -= prob*a*(a-1)/(a+b+c)/(a+b+c-1)

                if b > 1:
                    ndp[a][b-1][c] += prob*b*(b-1)/(a+b+c)/(a+b+c-1)
                    ndp[a][b][c] -= prob*b*(b-1)/(a+b+c)/(a+b+c-1)

                if c > 1:
                    ndp[a][b][c-1] += prob*c*(c-1)/(a+b+c)/(a+b+c-1)
                    ndp[a][b][c] -= prob*c*(c-1)/(a+b+c)/(a+b+c-1)
                ndp[a][b][c] += prob
    dp = ndp
a = 0
b = 0
c = 0
for i in range(A+1):
    for j in range(B+1):
        for k in range(C+1):
            prob = dp[i][j][k]
            a += (A-i)*prob
            b += (B-j)*prob
            c += (C-k)*prob
print(a,b,c)
0