結果

問題 No.728 ギブ and テイク
コンテスト
ユーザー kopricky
提出日時 2018-08-07 03:32:45
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,004 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 308 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 119,504 KB
最終ジャッジ日時 2024-09-19 18:24:26
合計ジャッジ時間 26,660 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 29 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
import bisect


class BIT:
    def __init__(self, node_size):
        self._node = node_size+1
        self.bit = [0]*self._node

    def add(self, index, add_val):
        while index < self._node:
            self.bit[index] += add_val
            index += index & -index

    def sum(self, index):
        res = 0
        while index > 0:
            res += self.bit[index]
            index -= index & -index
        return res


if __name__ == '__main__':
    n = int(sys.stdin.readline())
    a = tuple(map(int, sys.stdin.readline().split()))
    l, r = [0]*n, [0]*n
    vec = [None]*(2*n)
    for i in range(n):
        l[i], r[i] = map(int, sys.stdin.readline().split())
        vec[i], vec[n+i] = (a[i], n+i), (a[i]-l[i], i)
    vec.sort()
    bt = BIT(n)
    ans = 0
    for i in range(2*n):
        q = vec[i][1]
        if q >= n:
            q -= n
            ans += bt.sum(bisect.bisect_right(a, a[q]+r[q])) - bt.sum(q+1)
        else:
            bt.add(q+1, 1)
    print(ans)
0