結果

問題 No.1071 ベホマラー
ユーザー FromBooska
提出日時 2023-04-18 12:22:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 214 ms / 2,000 ms
コード長 1,365 bytes
コンパイル時間 1,014 ms
コンパイル使用メモリ 82,128 KB
実行使用メモリ 91,520 KB
最終ジャッジ日時 2024-10-13 10:44:17
合計ジャッジ時間 5,122 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

# ベホイミをa回、ベホマラーをb回使うとする、そのときのコストaX+bYを最小化したい
# bを決めればaは決まる
# ということはコストはbの関数となる、まずそれを実装しよう

N, K, X, Y = map(int, input().split())
A = list(map(int, input().split()))
A = [t-1 for t in A]

def find_a(b):
    a = 0
    for i in range(N):
        calc = max(0, (A[i]-b*K+K-1)//K)
        a += calc
    return a

maxA = max(A)
b_max = (maxA+K-1)//K

def cost(b):
    a = find_a(b)
    return a*X+b*Y
    

if b_max == 0:
    print(cost(0))
    exit()


# 谷、つまり1極値を求める三分探索での解き方
# https://roiti46.hatenablog.com/entry/2015/04/29/yukicoder_No.198_%E3%82%AD%E3%83%A3%E3%83%B3%E3%83%87%E3%82%A3%E3%83%BC%E3%83%BB%E3%83%9C%E3%83%83%E3%82%AF%E3%82%B9%EF%BC%92
# https://qiita.com/ganyariya/items/1553ff2bf8d6d7789127

# 範囲に注意、ギリギリありえない値とする
left, right = -1, b_max+1
while right-left > 2:
    left_mid, right_mid = (2*left+right)//3, (left+2*right)//3
    if cost(left_mid) <= cost(right_mid):
        # 右midの方が大きいから右を右midに変える
        right = right_mid
    else:
        left = left_mid

#print(left_mid, right_mid)

ans = min(cost(left_mid), cost(right_mid))
# leftとrightの間も入れる必要あるのか?
print(ans)


0