結果

問題 No.1012 荷物収集
ユーザー kkyo
提出日時 2020-05-04 19:47:22
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 877 ms / 2,000 ms
コード長 902 bytes
コンパイル時間 101 ms
コンパイル使用メモリ 12,928 KB
実行使用メモリ 45,104 KB
最終ジャッジ日時 2024-06-24 21:34:21
合計ジャッジ時間 22,177 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

# Xi < X < Xi+1 となる X 地点に移動させるためのコスト
# Σ{i=0~n}|Xi-X|*Wi
# = Σ{i=0,k}(X-Xi)*Wi + Σ{i=k+1,n}(Xi-X)*Wi
# = X*Σ{i=0~k}(Wi) - Σ{i=0~k}(XiWi) + Σ{i=k+1~n}(XiWi) - X*Σ{i=k+1~n}(Wi)
# = 2*X*Σ{i=0~k}(Wi) - X*Σ{i=0~n}(Wi) - 2*Σ{i=0~k}(XiWi) + Σ{i=0~n}(XiWi)
#           A                B                     C                  D

import bisect

N, Q = map(int, input().split())
bags = []
for _ in range(N):
    x, w = map(int, input().split())
    bags.append((x, w))
bags.sort()

X, W = [], []
for x, w in bags:
    X.append(x)
    W.append(w)

A, C = [0] * (N + 1), [0] * (N + 1)
B, D = 0, 0
for i in range(1, N + 1):
    A[i] = A[i - 1] + W[i - 1]
    C[i] = C[i - 1] + X[i - 1] * W[i - 1]
    B += W[i - 1]
    D += X[i - 1] * W[i - 1]

for q in [int(s) for s in input().split()]:
    k = bisect.bisect_left(X, q)
    print(2 * q * A[k] - q * B - 2 * C[k] + D)
0