結果

問題 No.1688 Veterinarian
ユーザー kohei2019kohei2019
提出日時 2023-06-04 21:39:10
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 194 ms / 3,000 ms
コード長 931 bytes
コンパイル時間 355 ms
コンパイル使用メモリ 87,272 KB
実行使用メモリ 79,640 KB
最終ジャッジ日時 2023-08-28 08:29:34
合計ジャッジ時間 3,127 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 74 ms
71,400 KB
testcase_01 AC 68 ms
71,396 KB
testcase_02 AC 72 ms
71,248 KB
testcase_03 AC 72 ms
71,104 KB
testcase_04 AC 72 ms
71,504 KB
testcase_05 AC 71 ms
71,376 KB
testcase_06 AC 97 ms
76,756 KB
testcase_07 AC 93 ms
76,784 KB
testcase_08 AC 194 ms
79,640 KB
testcase_09 AC 192 ms
79,264 KB
testcase_10 AC 125 ms
77,944 KB
testcase_11 AC 123 ms
78,040 KB
testcase_12 AC 88 ms
76,232 KB
testcase_13 AC 98 ms
77,700 KB
testcase_14 AC 96 ms
77,456 KB
testcase_15 AC 119 ms
77,796 KB
testcase_16 AC 83 ms
76,036 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

A,B,C,N = map(int,input().split())
dp = [[[0]*(C+1) for i in range(B+1)] for j in range(A+1)]
dp[A][B][C] = 1
# a,b,c枚あるとき、全パターン(a+b+c)C2,a2枚はaC2,b2枚はbC2,c2枚はcC2

for i in range(N):
    for j in range(1,A+1):
        for k in range(1,B+1):
            for l in range(1,C+1):
                if dp[j][k][l]==0:
                    continue
                alln = (j+k+l)*(j+k+l-1)//2
                dp[j-1][k][l] += dp[j][k][l]*(j)*(j-1)/2/alln
                dp[j][k-1][l] += dp[j][k][l]*(k)*(k-1)/2/alln
                dp[j][k][l-1] += dp[j][k][l]*(l)*(l-1)/2/alln
                dp[j][k][l] -= dp[j][k][l]*((j)*(j-1)/2/alln+(k)*(k-1)/2/alln+(l)*(l-1)/2/alln)

Av = 0
Bv = 0
Cv = 0


for j in range(1,A+1):
    for k in range(1,B+1):
        for l in range(1,C+1):
            Av += (A-j)*dp[j][k][l]
            Bv += (B-k)*dp[j][k][l]
            Cv += (C-l)*dp[j][k][l]

print(*[Av,Bv,Cv])
0