結果

問題 No.1982 [Cherry 4th Tune B] 絶険
コンテスト
ユーザー pitP
提出日時 2022-06-19 08:26:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,273 ms / 3,000 ms
コード長 2,085 bytes
コンパイル時間 518 ms
コンパイル使用メモリ 82,180 KB
実行使用メモリ 222,220 KB
最終ジャッジ日時 2024-10-11 04:55:55
合計ジャッジ時間 26,048 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

#先頭からのsum,addをO(logN)
class Fenwick_Tree:
    def __init__(self, n):
        self._n = n  # 要素数
        self.data = [0] * n
        self.depth = n.bit_length()

    def add(self, p, x):
        assert 0 <= p < self._n
        p += 1  # 0-indexed -> 1-indexed
        while p <= self._n:
            self.data[p - 1] += x # dataを更新
            p += p & -p  # pにLSB(p)を加算,data[i]の長さを示す
    
    def _sum(self, r):#s = a0+a1+...+a[r-1] or 0
        s = 0  # 総和を入れる変数
        while r > 0:
            s += self.data[r - 1]
            r -= r & -r  # rからLSB(r)を減算
        return s
    
    def sum(self, l, r):#[l,r)の総和
        assert 0 <= l <= r <= self._n
        return self._sum(r) - self._sum(l)

    def bisect_l(self,x): #wa >= xとなる最小のidx
        pos = 0 ; wa = 0
        for i in range(self.depth,-1,-1):
            k = pos + (1<<i)
            if k < self._n and wa + self.data[k-1] < x:
                wa += self.data[k-1]
                pos += (1<<i)
        return pos 
    
    def bisect_r(self,x): #wa >= xとなる最大のidx
        if x < 0 : return 0
        pos = 0 ; wa = 0
        for i in range(self.depth+1,-1,-1):
            k = pos + (1 << i)
            if k <= self._n and wa + self.data[k-1] <= x:
                wa += self.data[k-1]
                pos += (1<<i)
        return pos

n,k,q = map(int,input().split())
lrch = [list(map(int,input().split())) for _ in range(k)]
ix = [list(map(int,input().split())) for _ in range(q)]

Q = [[] for _ in range(n+2)]
for idx , (l,_,_,h) in enumerate(lrch):
    Q[l].append([True,h,idx])
for ans_idx , (i,x) in enumerate(ix):
    Q[i].append([False,ans_idx,x])
for idx , (_,r,_,h) in enumerate(lrch):
    Q[r].append([True,-h,idx])


ans = [-1] * q
fw = Fenwick_Tree(k+2)
for i in range(n+2):
    for boo,x,y in Q[i]:
        if boo : fw.add(y,x)
        else :
            idx = fw.bisect_l(y)
            if idx >= k : continue
            ans[x] = lrch[idx][2]

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