結果
| 問題 |
No.1802 Range Score Query for Bracket Sequence
|
| コンテスト | |
| ユーザー |
tktk_snsn
|
| 提出日時 | 2022-01-07 22:47:53 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 509 ms / 2,000 ms |
| コード長 | 2,027 bytes |
| コンパイル時間 | 218 ms |
| コンパイル使用メモリ | 82,304 KB |
| 実行使用メモリ | 94,068 KB |
| 最終ジャッジ日時 | 2024-06-30 02:17:32 |
| 合計ジャッジ時間 | 9,015 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | AC * 24 |
ソースコード
import sys
input = sys.stdin.readline
class FenwickTree(object):
def __init__(self, n):
self.n = n
self.log = n.bit_length()
self.data = [0] * n
self.raw = [0] * n
def __sum(self, r):
s = 0
while r > 0:
s += self.data[r - 1]
r -= r & -r
return s
def get(self, i):
return self.raw[i]
def add(self, p, x):
""" a[p] += xを行う"""
self.raw[p] += x
p += 1
while p <= self.n:
self.data[p - 1] += x
p += p & -p
def update(self, p, x):
self.add(p, x - self.get(p))
def sum(self, l, r):
"""a[l] + a[l+1] + .. + a[r-1]を返す"""
return self.__sum(r) - self.__sum(l)
def lower_bound(self, x):
"""a[0] + a[1] + .. a[i] >= x となる最小のiを返す"""
if x <= 0:
return -1
i = 0
k = 1 << self.log
while k:
if i + k <= self.n and self.data[i + k - 1] < x:
x -= self.data[i + k - 1]
i += k
k >>= 1
return i
def __repr__(self):
res = [self.sum(i, i+1) for i in range(self.n)]
return " ".join(map(str, res))
N, Q = map(int, input().split())
S = list(input().rstrip())
bit = FenwickTree(N + 1)
for i in range(N-1):
if S[i] == ")" and S[i+1] == "(":
bit.add(i, 1)
def swap(i):
S[i] = (set("()") ^ set(S[i])).pop()
if i + 1 < N:
if S[i] == ")" and S[i+1] == "(":
bit.update(i, 1)
else:
bit.update(i, 0)
if i > 0:
if S[i-1] == ")" and S[i] == "(":
bit.update(i-1, 1)
else:
bit.update(i-1, 0)
for _ in range(Q):
flag, *arg = map(int, input().split())
if flag == 1:
swap(arg[0]-1)
else:
l, r = arg
l -= 1
r -= 1
V = bit.sum(l, r)
if S[l] == "(":
V += 1
if S[r] == ")":
V += 1
print(V - 1)
tktk_snsn