結果

問題 No.1688 Veterinarian
コンテスト
ユーザー U 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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

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