結果

問題 No.1688 Veterinarian
ユーザー kohei2019
提出日時 2023-06-04 21:39:10
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 173 ms / 3,000 ms
コード長 931 bytes
コンパイル時間 360 ms
コンパイル使用メモリ 82,136 KB
実行使用メモリ 77,948 KB
最終ジャッジ日時 2024-12-29 03:56:42
合計ジャッジ時間 2,421 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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
# a,b,c枚あるとき、全パターン(a+b+c)C2,a2枚はaC2,b2枚はbC2,c2枚はcC2

for i in range(N):
    for j in range(1,A+1):
        for k in range(1,B+1):
            for l in range(1,C+1):
                if dp[j][k][l]==0:
                    continue
                alln = (j+k+l)*(j+k+l-1)//2
                dp[j-1][k][l] += dp[j][k][l]*(j)*(j-1)/2/alln
                dp[j][k-1][l] += dp[j][k][l]*(k)*(k-1)/2/alln
                dp[j][k][l-1] += dp[j][k][l]*(l)*(l-1)/2/alln
                dp[j][k][l] -= dp[j][k][l]*((j)*(j-1)/2/alln+(k)*(k-1)/2/alln+(l)*(l-1)/2/alln)

Av = 0
Bv = 0
Cv = 0


for j in range(1,A+1):
    for k in range(1,B+1):
        for l in range(1,C+1):
            Av += (A-j)*dp[j][k][l]
            Bv += (B-k)*dp[j][k][l]
            Cv += (C-l)*dp[j][k][l]

print(*[Av,Bv,Cv])
0