結果

問題 No.1982 [Cherry 4th Tune B] 絶険
ユーザー tassei903
提出日時 2022-06-18 16:22:52
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,114 bytes
コンパイル時間 495 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 194,460 KB
最終ジャッジ日時 2024-12-28 06:23:45
合計ジャッジ時間 56,301 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 34 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
######################################################################## Binary Indexed Tree (Fenwick Tree)
# 0-indexed
class BIT:
    def __init__(self, n):
        self.n = n
        self.data = [0]*(n+1)
        self.el = [0]*(n+1)
    def sum(self, i):
        s = 0
        while i > 0:
            s += self.data[i]
            i -= i & -i
        return s
    def add(self, i, x):
        assert 0 <= i < self.n
        i = i+1
        self.el[i] += x
        while i <= self.n:
            self.data[i] += x
            i += i & -i
    def set(self, i, x):
        assert 0 <= i <= self.n
        self.add(i,x-self.get(i))
    def get(self, i, j=None):# j=Noneのときiを取得, [i, j)の和を取得
        if j is None:
            return self.el[i+1]
        return self.sum(j) - self.sum(i)
    def lower_bound(self, k):# a[0],…,a[i]>kなる最小のi
        n = self.n
        x = 0
        r = 1
        while r <= n:r <<= 1
        size = r
        while size:
            if x + size <= n and self.data[x+size] <= k:
                k-=self.data[x+size]
                x += size
            size >>= 1
        return x
    def debug(self):
        res = []
        n = self.n
        for i in range(n):
            res.append(self.get(i))
        print(*res)

N,K,Q = na()
eve = []
color = [-1]*K
for i in range(K):
    l,r,c,h = na()
    color[i] = c
    eve.append((l, 0, i, h))
    eve.append((r, 2, i, h))
ans = [-1]*Q
for j in range(Q):
    i, x = na()
    eve.append((i, 1, j, x-1))

eve = sorted(eve)
bit = BIT(K)
for i, p, j, h in eve:
    if p == 0:
        bit.add(j, h)
    elif p == 2:
        bit.add(j, -h)
    else:
        x = bit.lower_bound(h)
        #bit.debug()
        #print(x)
        if x == K:
            ans[j] = -1
        else:
            ans[j] = color[x]

print(*ans,sep="\n")
0