結果

問題 No.1688 Veterinarian
ユーザー roarisroaris
提出日時 2021-09-24 21:56:53
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 712 ms / 3,000 ms
コード長 1,102 bytes
コンパイル時間 2,481 ms
コンパイル使用メモリ 86,276 KB
実行使用メモリ 104,136 KB
最終ジャッジ日時 2023-09-18 21:17:30
合計ジャッジ時間 5,110 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 101 ms
71,304 KB
testcase_01 AC 96 ms
71,460 KB
testcase_02 AC 104 ms
75,812 KB
testcase_03 AC 97 ms
71,288 KB
testcase_04 AC 98 ms
71,540 KB
testcase_05 AC 97 ms
71,528 KB
testcase_06 AC 137 ms
77,532 KB
testcase_07 AC 180 ms
82,604 KB
testcase_08 AC 712 ms
104,136 KB
testcase_09 AC 652 ms
102,324 KB
testcase_10 AC 267 ms
81,124 KB
testcase_11 AC 267 ms
81,456 KB
testcase_12 AC 168 ms
79,368 KB
testcase_13 AC 188 ms
79,112 KB
testcase_14 AC 207 ms
80,620 KB
testcase_15 AC 235 ms
80,592 KB
testcase_16 AC 156 ms
78,856 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