結果

問題 No.1012 荷物収集
ユーザー brthyyjp
提出日時 2020-03-20 23:20:28
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,139 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 82,420 KB
実行使用メモリ 187,228 KB
最終ジャッジ日時 2024-12-15 15:47:14
合計ジャッジ時間 78,141 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 18 TLE * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

def main():
    n, q = map(int, input().split())
    L = []
    xs = []
    for i in range(n):
        x, w = map(int, input().split())
        L.append([x, w])
        xs.append(x)
    X = list(map(int, input().split()))
    L.sort()
    xs.sort()

    Wl = [0]*n
    Wr = [0]*n
    wsl = [0]*n
    wsr = [0]*n
    for i in range(n):
        if i == 0:
            wsl[i] = L[i][1]
        else:
            wsl[i] = wsl[i-1] + L[i][1]
            Wl[i] = wsl[i-1]*(xs[i]-xs[i-1]) + Wl[i-1]

    for i in reversed(range(n)):
        if i == n-1:
            wsr[i] = L[i][1]
        else:
            j = n-1-i
            wsr[i] = wsr[i+1] + L[i][1]
            Wr[i] = wsr[i+1]*(xs[i+1]-xs[i]) + Wr[i+1]
    #print(Wl)
    #print(Wr)
    import bisect
    for x in X:
        if x < min(xs):
            ans = wsr[0]*(xs[0]-x)+Wr[0]
        elif x >= max(xs):
            ans = wsl[-1]*(x-xs[-1])+Wl[-1]
        else:
            id = bisect.bisect_right(xs, x)
            ans = wsl[id-1]*(x-xs[id-1])+Wl[id-1]+ wsr[id]*(xs[id]-x)+Wr[id]
        print(ans)

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