結果

問題 No.1688 Veterinarian
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2021-09-24 22:12:52
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 250 ms / 3,000 ms
コード長 1,140 bytes
コンパイル時間 873 ms
コンパイル使用メモリ 81,788 KB
実行使用メモリ 89,488 KB
最終ジャッジ日時 2024-07-05 10:44:29
合計ジャッジ時間 2,375 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 36 ms
52,728 KB
testcase_01 AC 37 ms
52,964 KB
testcase_02 AC 37 ms
54,296 KB
testcase_03 AC 35 ms
53,224 KB
testcase_04 AC 35 ms
53,056 KB
testcase_05 AC 39 ms
53,088 KB
testcase_06 AC 48 ms
63,812 KB
testcase_07 AC 64 ms
68,840 KB
testcase_08 AC 250 ms
89,488 KB
testcase_09 AC 212 ms
89,080 KB
testcase_10 AC 99 ms
76,568 KB
testcase_11 AC 107 ms
77,352 KB
testcase_12 AC 49 ms
64,876 KB
testcase_13 AC 80 ms
76,444 KB
testcase_14 AC 68 ms
71,028 KB
testcase_15 AC 116 ms
77,156 KB
testcase_16 AC 53 ms
64,892 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