結果
問題 | No.1688 Veterinarian |
ユーザー |
|
提出日時 | 2021-09-24 21:34:03 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 241 ms / 3,000 ms |
コード長 | 1,223 bytes |
コンパイル時間 | 233 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 98,944 KB |
最終ジャッジ日時 | 2024-07-05 10:09:30 |
合計ジャッジ時間 | 2,596 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 14 |
ソースコード
import sys,random input = lambda :sys.stdin.readline().rstrip() mi = lambda :map(int,input().split()) li = lambda :list(mi()) A,B,C,N = mi() dp = [[[0 for k in range(C+1)] for j in range(B+1)] for i in range(A+1)] dp[A][B][C] = 1 for _ in range(N): ndp = [[[0 for k in range(C+1)] for j in range(B+1)] for i in range(A+1)] for i in range(A+1): for j in range(B+1): for k in range(C+1): if not dp[i][j][k]: continue S = i + j + k if S==0: ndp[i][j][k] += dp[i][j][k] continue base = dp[i][j][k]/(S*(S-1)//2) if i: ndp[i-1][j][k] += i*(i-1)//2 *base if j: ndp[i][j-1][k] += j*(j-1)//2 *base if k: ndp[i][j][k-1] += k*(k-1)//2 *base ndp[i][j][k] += (S*(S-1)//2-i*(i-1)//2-j*(j-1)//2-k*(k-1)//2) * base dp = ndp res = [0,0,0] for i in range(A+1): for j in range(B+1): for k in range(C+1): p = dp[i][j][k] res[0] += (A-i) * p res[1] += (B-j) * p res[2] += (C-k) * p print(*res)