結果

問題 No.2421 entersys?
ユーザー aaaaaaaaaa2230aaaaaaaaaa2230
提出日時 2023-08-12 14:29:47
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,659 ms / 3,000 ms
コード長 3,070 bytes
コンパイル時間 252 ms
コンパイル使用メモリ 82,224 KB
実行使用メモリ 316,588 KB
最終ジャッジ日時 2024-11-19 19:46:58
合計ジャッジ時間 25,983 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
53,412 KB
testcase_01 AC 52 ms
65,880 KB
testcase_02 AC 85 ms
77,612 KB
testcase_03 AC 44 ms
60,900 KB
testcase_04 AC 52 ms
65,920 KB
testcase_05 AC 68 ms
72,340 KB
testcase_06 AC 69 ms
73,776 KB
testcase_07 AC 71 ms
72,464 KB
testcase_08 AC 80 ms
77,432 KB
testcase_09 AC 98 ms
77,960 KB
testcase_10 AC 64 ms
70,600 KB
testcase_11 AC 1,263 ms
261,112 KB
testcase_12 AC 1,284 ms
261,612 KB
testcase_13 AC 1,236 ms
262,476 KB
testcase_14 AC 1,264 ms
262,456 KB
testcase_15 AC 1,256 ms
262,260 KB
testcase_16 AC 1,356 ms
263,148 KB
testcase_17 AC 1,354 ms
261,108 KB
testcase_18 AC 1,362 ms
263,184 KB
testcase_19 AC 1,374 ms
263,024 KB
testcase_20 AC 1,324 ms
261,204 KB
testcase_21 AC 1,226 ms
244,368 KB
testcase_22 AC 1,012 ms
253,076 KB
testcase_23 AC 1,659 ms
316,088 KB
testcase_24 AC 1,638 ms
315,964 KB
testcase_25 AC 1,643 ms
316,588 KB
testcase_26 AC 828 ms
247,840 KB
testcase_27 AC 835 ms
250,860 KB
testcase_28 AC 832 ms
256,448 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

class BIT:
    def __init__(self, n):
        self.size = n
        self.tree = [0]*(n+1)
 
    def build(self, list):
        self.tree[1:] = list.copy()
        for i in range(self.size+1):
            j = i + (i & (-i))
            if j < self.size+1:
                self.tree[j] += self.tree[i]

    def sum(self, i):
        # [0, i) の要素の総和を返す
        s = 0
        while i>0:
            s += self.tree[i]
            i -= i & -i
        return s
    # 0 index を 1 index に変更  転倒数を求めるなら1を足していく
    def add(self, i, x):
        i += 1
        while i <= self.size:
            self.tree[i] += x
            i += i & -i

    # 総和がx以上になる位置のindex をbinary search
    def bsearch(self,x):
        le = 0
        ri = 1<<(self.size.bit_length()-1)
        while ri > 0:
            if le+ri <= self.size and self.tree[le+ri]<x:
                x -= self.tree[le+ri]
                le += ri
            ri >>= 1
        return le+1


n = int(input())

member = {}
alls = set()
XLR = []
for i in range(n):
    x,l,r = input().split()
    l,r = int(l),int(r)
    if x not in member:
        member[x] = set()
    
    member[x].add(l)
    member[x].add(r+1)
    alls.add(l)
    alls.add(r+1)
    XLR.append([x,l,r])

q = int(input())
Q = []
for _ in range(q):
    l = input().split()
    if l[0] == "1":
        if l[1] not in member:
            member[l[1]] = set()
        l[0] = int(l[0])
        l[2] = int(l[2])
        member[l[1]].add(l[2])
        Q.append(l)

    elif l[0] == "2":
        l[0] = int(l[0])
        l[1] = int(l[1])
        alls.add(l[1])
        Q.append(l)
    else:
        if l[1] not in member:
            member[l[1]] = set()
        l[0] = int(l[0])
        l[2] = int(l[2])
        l[3] = int(l[3])
        member[l[1]].add(l[2])
        member[l[1]].add(l[3]+1)
        alls.add(l[2])
        alls.add(l[3]+1)
        Q.append(l)

keys = sorted(member.keys())
member_dic = {x:i for i,x in enumerate(keys)}
bits = [BIT(len(member[key])) for key in keys]
compress = []

for key in keys:
    comp_keys = {x:i for i,x in enumerate(sorted(member[key]))}
    compress.append(comp_keys)
all_coms = {x:i for i,x in enumerate(sorted(alls))}

all_bit = BIT(len(all_coms))
for x,l,r in XLR:
    num = member_dic[x]
    bits[num].add(compress[num][l],1)
    bits[num].add(compress[num][r+1],-1)
    all_bit.add(all_coms[l],1)
    all_bit.add(all_coms[r+1],-1)

for q in Q:
    # print("query",q)
    if q[0] == 1:
        # print("A")
        num = member_dic[q[1]]
        pos = compress[num][q[2]]
        if bits[num].sum(pos+1):
            print("Yes")
        else:
            print("No")

    elif q[0] == 2:
        # print("B")
        pos = all_coms[q[1]]
        print(all_bit.sum(pos+1))
    
    else:
        # print(q)
        # print("C")
        num = member_dic[q[1]]
        l,r = q[2:]
        all_bit.add(all_coms[l],1)
        all_bit.add(all_coms[r+1],-1)

        bits[num].add(compress[num][l],1)
        bits[num].add(compress[num][r+1],-1)

0