結果

問題 No.1251 絶対に間違ってはいけない最小化問題
ユーザー keijak
提出日時 2020-10-09 22:24:19
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,294 bytes
コンパイル時間 139 ms
コンパイル使用メモリ 82,000 KB
実行使用メモリ 114,676 KB
最終ジャッジ日時 2024-07-20 12:30:03
合計ジャッジ時間 21,556 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 12 WA * 30
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys

sys.setrecursionlimit(10 ** 8)
ini = lambda: int(sys.stdin.readline())
inl = lambda: [int(x) for x in sys.stdin.readline().split()]
ins = lambda: sys.stdin.readline().rstrip()
debug = lambda *a, **kw: print("\033[33m", *a, "\033[0m", **dict(file=sys.stderr, **kw))


def solve():
    n = ini()
    A = inl()
    B = inl()

    P = []
    for i in range(n):
        P.append((A[i], B[i]))
    P.sort()

    cum = [None] * (n + 1)
    cum[0] = 0
    for i in range(n):
        cum[i + 1] = cum[i] + P[i][1]

    mp = P[n - 1][0]
    for i in range(n):
        if cum[i] <= cum[n] - cum[i + 1] and cum[i + 1] > cum[n] - cum[i + 1]:
            mp = P[i][0]
            break
        if i < n - 1 and cum[i + 1] == cum[n] - cum[i + 1]:
            mp = (P[i][0] + P[i + 1][0]) * 0.5
            break

    def f(x):
        res = 0
        for i in range(n):
            res += B[i] * abs(x - A[i])
        return res

    # lo, hi, minima = min(A) - 1, max(A) + 1, None
    # for i in range(300):
    #     x = lo + (hi - lo) / 3.0
    #     y = lo + (hi - lo) * 2.0 / 3.0
    #     fx, fy = f(x), f(y)
    #     if fx > fy:
    #         lo = x
    #         minima = fy
    #     else:
    #         hi = y
    #         minima = fx
    return mp, int(f(mp) + 0.5)


print(*solve())
0