結果
問題 | No.1688 Veterinarian |
ユーザー | ygd. |
提出日時 | 2021-09-24 21:59:51 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
AC
|
実行時間 | 1,672 ms / 3,000 ms |
コード長 | 1,084 bytes |
コンパイル時間 | 79 ms |
コンパイル使用メモリ | 10,916 KB |
実行使用メモリ | 90,124 KB |
最終ジャッジ日時 | 2023-09-18 21:20:45 |
合計ジャッジ時間 | 8,019 ms |
ジャッジサーバーID (参考情報) |
judge13 / judge11 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 20 ms
8,812 KB |
testcase_01 | AC | 23 ms
8,760 KB |
testcase_02 | AC | 26 ms
8,904 KB |
testcase_03 | AC | 20 ms
8,872 KB |
testcase_04 | AC | 21 ms
8,672 KB |
testcase_05 | AC | 20 ms
8,848 KB |
testcase_06 | AC | 1,672 ms
90,124 KB |
testcase_07 | AC | 21 ms
8,840 KB |
testcase_08 | AC | 1,615 ms
89,876 KB |
testcase_09 | AC | 1,452 ms
84,380 KB |
testcase_10 | AC | 330 ms
25,676 KB |
testcase_11 | AC | 97 ms
12,952 KB |
testcase_12 | AC | 21 ms
8,748 KB |
testcase_13 | AC | 138 ms
15,944 KB |
testcase_14 | AC | 22 ms
8,700 KB |
testcase_15 | AC | 1,140 ms
70,228 KB |
testcase_16 | AC | 22 ms
8,832 KB |
ソースコード
import sys sys.setrecursionlimit(1000000) input = sys.stdin.buffer.readline from functools import lru_cache def comb2(x): return x*(x-1)//2 def main(): A,B,C,N = map(int,input().split()) @lru_cache(maxsize=None) def solve(a,b,c,n): if n == 0: return 0,0,0 if comb2(a+b+c) == 0: return 0,0,0 aget = 0 bget = 0 cget = 0 #aを1枚てにいれる確率 pa = comb2(a)/comb2(a+b+c) pb = comb2(b)/comb2(a+b+c) pc = comb2(c)/comb2(a+b+c) aget = pa * (1 + solve(a-1,b,c,n-1)[0]) + pb * solve(a,b-1,c,n-1)[0] + pc * solve(a,b,c-1,n-1)[0] + (1-(pa+pb+pc)) * solve(a,b,c,n-1)[0] bget = pa * solve(a-1,b,c,n-1)[1] + pb * (1 + solve(a,b-1,c,n-1)[1]) + pc * solve(a,b,c-1,n-1)[1] + (1-(pa+pb+pc)) * solve(a,b,c,n-1)[1] cget = pa * solve(a-1,b,c,n-1)[2] + pb * solve(a,b-1,c,n-1)[2] + pc * (1 + solve(a,b,c-1,n-1)[2]) + (1-(pa+pb+pc)) * solve(a,b,c,n-1)[2] return aget,bget,cget ans = solve(A,B,C,N) print(*ans) if __name__ == '__main__': main()