結果
問題 | No.1251 絶対に間違ってはいけない最小化問題 |
ユーザー | norioc |
提出日時 | 2024-06-20 05:16:41 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 382 ms / 2,000 ms |
コード長 | 635 bytes |
コンパイル時間 | 444 ms |
コンパイル使用メモリ | 82,012 KB |
実行使用メモリ | 110,124 KB |
最終ジャッジ日時 | 2024-06-20 05:17:03 |
合計ジャッジ時間 | 21,452 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
def ternary_search(f, lo, hi) -> float: l = lo r = hi for _ in range(100): c1 = (l * 2 + r) / 3 c2 = (l + r * 2) / 3 if f(c1) > f(c2): l = c1 else: r = c2 return l N = int(input()) A = list(map(int, input().split())) B = list(map(int, input().split())) def f(x): res = 0 for a, b in zip(A, B): res += b * abs(x - a) return res lo = min(A) hi = max(A) x = int(ternary_search(f, lo, hi)) x_ans = x v_ans = f(x_ans) for i in [x-2, x-1, x, x+1, x+2]: r = f(i) if v_ans > r: x_ans = i v_ans = r print(x_ans, v_ans)