結果

問題 No.1688 Veterinarian
ユーザー aaaaaaaaaa2230
提出日時 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

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