結果

問題 No.1688 Veterinarian
ユーザー U SU S
提出日時 2021-09-24 22:51:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 200 ms / 3,000 ms
コード長 1,681 bytes
コンパイル時間 312 ms
コンパイル使用メモリ 81,752 KB
実行使用メモリ 94,528 KB
最終ジャッジ日時 2024-07-05 11:07:10
合計ジャッジ時間 2,086 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 46 ms
61,692 KB
testcase_01 AC 46 ms
61,388 KB
testcase_02 AC 43 ms
62,456 KB
testcase_03 AC 43 ms
62,036 KB
testcase_04 AC 43 ms
61,480 KB
testcase_05 AC 44 ms
62,396 KB
testcase_06 AC 51 ms
67,712 KB
testcase_07 AC 55 ms
71,224 KB
testcase_08 AC 200 ms
94,528 KB
testcase_09 AC 185 ms
90,484 KB
testcase_10 AC 92 ms
78,052 KB
testcase_11 AC 96 ms
79,132 KB
testcase_12 AC 53 ms
68,708 KB
testcase_13 AC 74 ms
77,448 KB
testcase_14 AC 61 ms
73,036 KB
testcase_15 AC 103 ms
78,156 KB
testcase_16 AC 56 ms
69,564 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#import sys
# input = sys.stdin.readline
def mp():return map(int,input().split())
def lmp():return list(map(int,input().split()))
def mps(A):return [tuple(map(int, input().split())) for _ in range(A)]
import math
import bisect
import heapq
from copy import deepcopy as dc
from itertools import accumulate
from collections import Counter, defaultdict, deque
def ceil(U,V):return (U+V-1)//V
def modf1(N,MOD):return (N-1)%MOD+1
inf = int(1e20)
mod = int(1e9+7)

u = [i*(i-1)//2 for i in range(151)]
a,b,c,n = mp()
dp = [[[0]*(c+1) for i in range(b+1)] for j in range(a+1)]
dp[a][b][c] = 1
for kai in range(n):
    cnt = [[[0]*(c+1) for i in range(b+1)] for j in range(a+1)]
    for i in range(a+1):
        for j in range(b+1):
            for k in range(c+1):
                if dp[i][j][k] == 0:continue
                num = i+j+k
                num_prob = 1
                if i != 1:
                    prob = u[i]/u[num]
                    cnt[i-1][j][k] += dp[i][j][k]*prob
                    num_prob -= prob
                if j != 1:
                    prob = u[j]/u[num]
                    cnt[i][j-1][k] += dp[i][j][k]*prob
                    num_prob -= prob
                if k != 1:
                    prob = u[k]/u[num]
                    cnt[i][j][k-1] += dp[i][j][k]*prob
                    num_prob -= prob
                cnt[i][j][k] += dp[i][j][k]*num_prob
    dp = cnt
#print(dp)
ans = [0,0,0]
for i in range(a+1):
    for j in range(b+1):
        for k in range(c+1):
            if dp[i][j][k] == 0:continue
            ans[0] += dp[i][j][k]*(a-i)
            ans[1] += dp[i][j][k]*(b-j)
            ans[2] += dp[i][j][k]*(c-k)
print(*ans)
0