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