結果

問題 No.1251 絶対に間違ってはいけない最小化問題
コンテスト
ユーザー keijak
提出日時 2020-10-09 23:07:28
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,478 bytes
コンパイル時間 100 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 51,740 KB
最終ジャッジ日時 2024-07-20 14:02:32
合計ジャッジ時間 24,021 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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()

    if n == 1:
        return A[0], 0

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

    mp = 0
    for i in range(n):
        if cum[i] <= cum[n] - cum[i + 1] and cum[i + 1] > cum[n] - cum[i + 1]:
            mp = i
            break
        if i < n - 1 and cum[i + 1] == cum[n] - cum[i + 1]:
            mp = i
            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
    if mp == n - 1:
        return P[mp][0], int(f(P[mp][0]) + 0.5)
    if f(P[mp][0]) <= f(P[mp + 1][0]):
        return P[mp][0], int(f(P[mp][0]) + 0.5)
    else:
        return P[mp + 1][0], int(f(P[mp + 1][0]) + 0.5)


print(*solve())
0