結果
問題 |
No.1012 荷物収集
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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)