結果

問題 No.1688 Veterinarian
ユーザー U SU S
提出日時 2021-09-24 22:51:15
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 301 ms / 3,000 ms
コード長 1,681 bytes
コンパイル時間 436 ms
コンパイル使用メモリ 87,108 KB
実行使用メモリ 98,552 KB
最終ジャッジ日時 2023-09-18 21:56:14
合計ジャッジ時間 3,934 ms
ジャッジサーバーID
（参考情報） 		judge12 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 120 ms
73,756 KB
testcase_01 AC 118 ms
73,792 KB
testcase_02 AC 120 ms
73,840 KB
testcase_03 AC 117 ms
73,740 KB
testcase_04 AC 115 ms
73,580 KB
testcase_05 AC 117 ms
73,588 KB
testcase_06 AC 128 ms
79,188 KB
testcase_07 AC 132 ms
78,992 KB
testcase_08 AC 301 ms
98,552 KB
testcase_09 AC 285 ms
96,468 KB
testcase_10 AC 174 ms
81,776 KB
testcase_11 AC 176 ms
82,784 KB
testcase_12 AC 129 ms
79,132 KB
testcase_13 AC 153 ms
81,364 KB
testcase_14 AC 141 ms
79,364 KB
testcase_15 AC 189 ms
82,644 KB
testcase_16 AC 134 ms
78,888 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