結果
問題 | No.1688 Veterinarian |
ユーザー | titia |
提出日時 | 2021-09-24 23:08:46 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,959 ms / 3,000 ms |
コード長 | 804 bytes |
コンパイル時間 | 274 ms |
コンパイル使用メモリ | 87,060 KB |
実行使用メモリ | 237,884 KB |
最終ジャッジ日時 | 2023-09-18 22:15:01 |
合計ジャッジ時間 | 10,600 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 107 ms
72,520 KB |
testcase_01 | AC | 130 ms
77,056 KB |
testcase_02 | AC | 177 ms
80,020 KB |
testcase_03 | AC | 107 ms
72,528 KB |
testcase_04 | AC | 109 ms
72,420 KB |
testcase_05 | AC | 108 ms
72,220 KB |
testcase_06 | AC | 1,959 ms
237,884 KB |
testcase_07 | AC | 107 ms
72,420 KB |
testcase_08 | AC | 1,910 ms
237,832 KB |
testcase_09 | AC | 1,766 ms
223,360 KB |
testcase_10 | AC | 508 ms
113,792 KB |
testcase_11 | AC | 257 ms
86,904 KB |
testcase_12 | AC | 107 ms
72,452 KB |
testcase_13 | AC | 309 ms
90,916 KB |
testcase_14 | AC | 121 ms
77,092 KB |
testcase_15 | AC | 1,410 ms
195,224 KB |
testcase_16 | AC | 137 ms
77,676 KB |
ソースコード
import sys input = sys.stdin.readline A,B,C,N=map(int,input().split()) def c2(x): return x*(x-1)//2 def plus(A,B): ANS=[0,0,0] for i in range(3): ANS[i]=A[i]+B[i] return ANS def mul(k,A): ANS=[0,0,0] for i in range(3): ANS[i]=k*A[i] return ANS from functools import lru_cache @lru_cache(maxsize=None) def calc(A,B,C,N): if N==0: return [0,0,0] ALL=c2(A+B+C) if ALL==0: return [0,0,0] ANS=[0,0,0] ANS=plus(ANS,mul(c2(A)/ALL,plus([1,0,0],calc(A-1,B,C,N-1)))) ANS=plus(ANS,mul(c2(B)/ALL,plus([0,1,0],calc(A,B-1,C,N-1)))) ANS=plus(ANS,mul(c2(C)/ALL,plus([0,0,1],calc(A,B,C-1,N-1)))) ANS=plus(ANS,mul((ALL-c2(A)-c2(B)-c2(C))/ALL,calc(A,B,C,N-1))) return ANS print(*calc(A,B,C,N))