結果

問題 No.1739 Princess vs. Dragoness (& AoE)
コンテスト
ユーザー 👑 Kazun
提出日時 2021-11-12 21:55:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,817 ms / 3,000 ms
コード長 1,051 bytes
コンパイル時間 159 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 121,260 KB
最終ジャッジ日時 2024-11-25 18:16:48
合計ジャッジ時間 24,287 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

def General_Binary_Increase_Search_Integer(L,R,cond,default=None):
    """条件式が単調増加であるとき, 整数上で二部探索を行う.
    L: 解の下限
    R: 解の上限
    cond: 条件(1変数関数, 広義単調増加を満たす)
    default: Lで条件を満たさないときの返り値
    """
    if not(cond(R)): return default

    if cond(L): return L

    R+=1
    while R-L>1:
        C=L+(R-L)//2
        if cond(C): R=C
        else: L=C
    return R
#==================================================
from heapq import heapify,heappop,heappush

def checks(K,N,A,B,X,Y,H):
    HH=[-(hp-K) for hp in H if hp>K]
    heapify(HH)

    for _ in range(A):
        if not HH:
            continue

        hp=-heappop(HH)
        if hp>=X:
            heappush(HH,-(hp-X))
    return -sum(HH)<=Y*B
#==================================================
N,A,B,X,Y=map(int,input().split())
H=list(map(int,input().split()))

check=lambda K:checks(K,N,A,B,X,Y,H)
print(General_Binary_Increase_Search_Integer(0,max(H),check,-1))
0