結果

問題 No.1012 荷物収集
ユーザー hedwig100
提出日時 2020-04-14 14:34:58
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 618 ms / 2,000 ms
コード長 1,004 bytes
コンパイル時間 380 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 54,732 KB
最終ジャッジ日時 2024-10-01 16:59:45
合計ジャッジ時間 17,133 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

INF = 10 ** 7
import sys
input = sys.stdin.readline
sys.setrecursionlimit(100000000)
dy = (-1,0,1,0)
dx = (0,1,0,-1)
from bisect import bisect_left

def main():
    n,q = map(int,input().split())
    work = [tuple(map(int,input().split())) for _ in range(n)]
    
    X = list(map(int,input().split()))
    work.sort(key = lambda x:x[0])
    R = [work[i][0] for i in range(n)]
    W = [work[i][1] for i in range(n)]

    cumlw = [0] * (n + 1)
    cumrw = [0] * (n + 1)
    cumlwx = [0] * (n + 1)
    cumrwx = [0] * (n + 1)

    for i in range(n):
        cumlw[i + 1] = cumlw[i] + W[i]
        cumlwx[i + 1] = cumlwx[i] + W[i] * R[i]
    
    for i in range(n - 1,-1,-1):
        cumrw[i] = cumrw[i + 1] + W[i]
        cumrwx[i] = cumrwx[i + 1] + W[i] * R[i]
    
    ans = []
    for i in range(q):
        idx = bisect_left(R,X[i])
        ret = X[i] * (cumlw[idx] - cumrw[idx]) - cumlwx[idx] + cumrwx[idx]
        ans.append(ret)
    
    print(*ans,sep = '\n')

if __name__ == '__main__':
    main()
0