結果

問題 No.1688 Veterinarian
ユーザー roarisroaris
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
53,888 KB
testcase_01 AC 40 ms
54,400 KB
testcase_02 AC 46 ms
62,208 KB
testcase_03 AC 36 ms
54,272 KB
testcase_04 AC 34 ms
53,760 KB
testcase_05 AC 37 ms
54,912 KB
testcase_06 AC 55 ms
72,832 KB
testcase_07 AC 107 ms
79,744 KB
testcase_08 AC 540 ms
102,920 KB
testcase_09 AC 517 ms
99,216 KB
testcase_10 AC 176 ms
79,592 KB
testcase_11 AC 183 ms
79,776 KB
testcase_12 AC 99 ms
78,320 KB
testcase_13 AC 114 ms
77,980 KB
testcase_14 AC 131 ms
79,540 KB
testcase_15 AC 159 ms
78,808 KB
testcase_16 AC 113 ms
77,952 KB
権限があれば一括ダウンロードができます

ソースコード

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