結果

問題 No.1251 絶対に間違ってはいけない最小化問題
ユーザー keijakkeijak
提出日時 2020-10-09 23:06:12
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
RE  
実行時間 -
コード長 1,475 bytes
コンパイル時間 334 ms
コンパイル使用メモリ 10,960 KB
実行使用メモリ 51,764 KB
最終ジャッジ日時 2023-09-27 19:29:21
合計ジャッジ時間 28,226 ms
ジャッジサーバーID
(参考情報)
judge12 / judge13
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 16 ms
8,164 KB
testcase_01 RE -
testcase_02 RE -
testcase_03 RE -
testcase_04 RE -
testcase_05 RE -
testcase_06 RE -
testcase_07 RE -
testcase_08 RE -
testcase_09 RE -
testcase_10 RE -
testcase_11 RE -
testcase_12 RE -
testcase_13 RE -
testcase_14 RE -
testcase_15 RE -
testcase_16 RE -
testcase_17 RE -
testcase_18 RE -
testcase_19 RE -
testcase_20 RE -
testcase_21 RE -
testcase_22 RE -
testcase_23 RE -
testcase_24 RE -
testcase_25 RE -
testcase_26 RE -
testcase_27 RE -
testcase_28 RE -
testcase_29 RE -
testcase_30 RE -
testcase_31 RE -
testcase_32 RE -
testcase_33 RE -
testcase_34 RE -
testcase_35 RE -
testcase_36 RE -
testcase_37 RE -
testcase_38 RE -
testcase_39 RE -
testcase_40 RE -
testcase_41 RE -
testcase_42 RE -
testcase_43 RE -
testcase_44 RE -
権限があれば一括ダウンロードができます

ソースコード

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]):
        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