結果

問題 No.703 ゴミ拾い Easy
ユーザー 山本信二
提出日時 2022-04-29 19:15:27
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,212 bytes
コンパイル時間 344 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 70,736 KB
最終ジャッジ日時 2024-06-29 00:31:12
合計ジャッジ時間 5,747 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 4
other AC * 4 WA * 16 TLE * 2 -- * 24
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input())
*A, = map(int, input().split())
*X, = map(int, input().split())
*Y, = map(int, input().split())


N0 = 2**(N-1).bit_length()
A = A + [10**6] * (N0 - N)
data = [None]*(2*N0+1)

# Li Chao Tree
def f(line, x):
    p, q = line
    return p*x + q

def _add_line(line, k, l, r):
    m = (l + r) // 2
    if data[k] is None:
        data[k] = line
        return
    lx = A[l]
    mx = A[m]
    rx = A[r-1]
    left = (f(line, lx) < f(data[k], lx))
    mid = (f(line, mx) < f(data[k], mx))
    right = (f(line, rx) < f(data[k], rx))
    if left and right:
        data[k] = line
        return
    if not left and not right:
        return
    if mid:
        data[k], line = line, data[k]
    if left != mid:
        _add_line(line, 2*k+1, l, m)
    else:
        _add_line(line, 2*k+2, m, r)

def add_line(line):
    return _add_line(line, 0, 0, N0)

def query(k, x):
    def gen(k, x):
        k += N0-1
        while k >= 0:
            if data[k]:
                yield f(data[k], x)
            k = (k - 1) // 2
    return min(gen(k, x))
dp = [0] * (N+1)
for i in range(N):
    x = X[i]; a = A[i]; y = Y[i]
    add_line((-2*x, x**2 + y**2 + dp[i]))
    dp[i+1] = query(i, a) + a**2
print(dp[N-1])
0