結果
問題 | No.412 花火大会 |
ユーザー |
|
提出日時 | 2016-08-12 23:24:01 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 32 ms / 2,000 ms |
コード長 | 1,595 bytes |
コンパイル時間 | 854 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 11,008 KB |
最終ジャッジ日時 | 2024-11-07 16:35:11 |
合計ジャッジ時間 | 1,623 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 18 |
ソースコード
import math B, C, D = map(int, input().split()) N = int(input()) Es = list(map(int, input().split())) def nCb(n, b): if b > n: return 0 return math.factorial(n)//math.factorial(b)//math.factorial(n-b) def solve(B, C, D, N, Es): Es.sort() bcd = [B, C, D] bcd.sort() b, c, d = bcd nb = len([e for e in Es if e >= b]) nc = len([e for e in Es if e >= c]) nd = len([e for e in Es if e >= d]) ans = 1 << N if b == c == d: ans -= (nCb(nb, 0) + nCb(nb, 1) + nCb(nb, 2)) * (1 << (N - nb)) # b 以上を0, 1, 2 個持っていく elif b == c: ans -= (1 << (N - nd)) # d以上を0個持っていく ans -= (1 << (N - nb)) * (1 + nb - nd) * nd # d以上を1個、b以上d未満を0個or1個持っていく ans -= (1 << (N - nb)) * 1 * (nCb(nd, 2)) # d以上を2個、b以上d未満を0個持っていく elif c == d: ans -= (1 << (N - nd)) * (1 + nd) # d以上を0個または1個持っていく ans -= (1 << (N - nb)) * (nCb(nd, 2)) # d以上のを2個、b以上d未満を0個持っていく else: ans -= (1 << (N - nd)) * 1 # d 以上を0個持っていく ans -= (1 << (N - nc)) * 1 * nd # d以上を1個、c以上d未満を0個持っていく ans -= (1 << (N - nb)) * 1 * (nc - nd) * nd # d以上を1個、c以上d未満を1個、b以上c未満を0個持っていく ans -= (1 << (N - nb)) * 1 * nCb(nd, 2) # d以上を2個、b以上d未満を0個持っていく return ans print(solve(B, C, D, N, Es))