結果

問題 No.1012 荷物収集
ユーザー brthyyjp
提出日時 2020-03-25 07:24:13
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 407 ms / 2,000 ms
コード長 753 bytes
コンパイル時間 393 ms
コンパイル使用メモリ 82,552 KB
実行使用メモリ 112,656 KB
最終ジャッジ日時 2025-01-01 19:49:00
合計ジャッジ時間 12,228 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
input = sys.stdin.readline

n, q = map(int, input().split())
L = [[0, 0] for _ in range(n)]
xs = [0]*n
for i in range(n):
    x, w = map(int, input().split())
    L[i][0] = x
    L[i][1] = w
    xs[i] = x
X = list(map(int, input().split()))

L.sort()
xs.sort()

ws = [0]*n
wxs = [0]*n
for i in range(n):
    x, w = L[i]
    ws[i] = w
    wxs[i] = x*w

cumw = [0]+ws
cumxw = [0]+wxs
from itertools import accumulate
cumw = list(accumulate(cumw))
cumxw = list(accumulate(cumxw))

import bisect
for x in X:
    k = bisect.bisect_right(xs, x)
    if k == 0:
        ans = cumxw[-1] - cumw[-1]*x
    elif k == n:
        ans = -cumxw[-1] + cumw[-1]*x
    else:
        ans = x*(cumw[-1]-2*(cumw[-1]-cumw[k]))+(cumxw[-1]-2*cumxw[k])
    print(ans)
0