結果
問題 |
No.1071 ベホマラー
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
# ベホイミを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)