結果

問題 No.1688 Veterinarian
ユーザー roaris
提出日時 2021-09-24 21:56:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 540 ms / 3,000 ms
コード長 1,102 bytes
コンパイル時間 156 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 102,920 KB
最終ジャッジ日時 2024-07-05 10:32:01
合計ジャッジ時間 3,334 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline
from collections import *

A, B, C, N = map(int, input().split())
dp = [[[0]*(C+1) for _ in range(B+1)] for _ in range(A+1)]
dp[A][B][C] = 1
f = lambda x: x*(x-1)//2

for _ in range(N):
    ndp = [[[0]*(C+1) for _ in range(B+1)] for _ in range(A+1)]
    
    for a in range(A+1):
        for b in range(B+1):
            for c in range(C+1):
                if a+b+c<2:
                    continue
                
                if a>1:
                    ndp[a-1][b][c] += dp[a][b][c]*f(a)/f(a+b+c)
                
                if b>1:
                    ndp[a][b-1][c] += dp[a][b][c]*f(b)/f(a+b+c)
                
                if c>1:
                    ndp[a][b][c-1] += dp[a][b][c]*f(c)/f(a+b+c)
                
                ndp[a][b][c] += dp[a][b][c]*(f(a+b+c)-f(a)-f(b)-f(c))/f(a+b+c)
    
    dp = ndp

pa, pb, pc = 0, 0, 0

for a in range(A+1):
    for b in range(B+1):
        for c in range(C+1):
            pa += dp[a][b][c]*(A-a)
            pb += dp[a][b][c]*(B-b)
            pc += dp[a][b][c]*(C-c)
            
print(pa, pb, pc)
0