結果
問題 | No.2933 Range ROT Query |
ユーザー |
👑 |
提出日時 | 2024-09-06 09:46:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,508 ms / 3,000 ms |
コード長 | 9,138 bytes |
コンパイル時間 | 860 ms |
コンパイル使用メモリ | 82,292 KB |
実行使用メモリ | 104,116 KB |
最終ジャッジ日時 | 2024-10-02 00:26:50 |
合計ジャッジ時間 | 54,100 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 50 |
ソースコード
sigma = 26class lazy_segtree:def update(self, k):self.d[k] = self.op(self.d[2 * k], self.d[2 * k + 1])def all_apply(self, k, f):self.d[k] = self.mapping(f, self.d[k])if k < self.size:self.lz[k] = self.composition(f, self.lz[k])def push(self, k):self.all_apply(2 * k, self.lz[k])self.all_apply(2 * k + 1, self.lz[k])self.lz[k] = self.identitydef __init__(self, V, OP, E, MAPPING, COMPOSITION, ID):self.n = len(V)self.log = (self.n - 1).bit_length()self.size = 1 << self.logself.d = [E for i in range(2 * self.size)]self.lz = [ID for i in range(self.size)]self.e = Eself.op = OPself.mapping = MAPPINGself.composition = COMPOSITIONself.identity = IDfor i in range(self.n):self.d[self.size + i] = V[i]for i in range(self.size - 1, 0, -1):self.update(i)def set(self, p, x):assert 0 <= p and p < self.np += self.sizefor i in range(self.log, 0, -1):self.push(p >> i)self.d[p] = xfor i in range(1, self.log + 1):self.update(p >> i)def get(self, p):assert 0 <= p and p < self.np += self.sizefor i in range(self.log, 0, -1):self.push(p >> i)return self.d[p]def prod(self, l, r):assert 0 <= l and l <= r and r <= self.nif l == r:return self.el += self.sizer += self.sizefor i in range(self.log, 0, -1):if ((l >> i) << i) != l:self.push(l >> i)if ((r >> i) << i) != r:self.push(r >> i)sml, smr = self.e, self.ewhile l < r:if l & 1:sml = self.op(sml, self.d[l])l += 1if r & 1:r -= 1smr = self.op(self.d[r], smr)l >>= 1r >>= 1return self.op(sml, smr)def all_prod(self):return self.d[1]def apply_point(self, p, f):assert 0 <= p and p < self.np += self.sizefor i in range(self.log, 0, -1):self.push(p >> i)self.d[p] = self.mapping(f, self.d[p])for i in range(1, self.log + 1):self.update(p >> i)def apply(self, l, r, f):assert 0 <= l and l <= r and r <= self.nif l == r:returnl += self.sizer += self.sizefor i in range(self.log, 0, -1):if ((l >> i) << i) != l:self.push(l >> i)if ((r >> i) << i) != r:self.push((r - 1) >> i)l2, r2 = l, rwhile l < r:if l & 1:self.all_apply(l, f)l += 1if r & 1:r -= 1self.all_apply(r, f)l >>= 1r >>= 1l, r = l2, r2for i in range(1, self.log + 1):if ((l >> i) << i) != l:self.update(l >> i)if ((r >> i) << i) != r:self.update((r - 1) >> i)def max_right(self, l, g):assert 0 <= l and l <= self.nassert g(self.e)if l == self.n:return self.nl += self.sizefor i in range(self.log, 0, -1):self.push(l >> i)sm = self.ewhile 1:while l % 2 == 0:l >>= 1if not (g(self.op(sm, self.d[l]))):while l < self.size:self.push(l)l = 2 * lif g(self.op(sm, self.d[l])):sm = self.op(sm, self.d[l])l += 1return l - self.sizesm = self.op(sm, self.d[l])l += 1if (l & -l) == l:breakreturn self.ndef min_left(self, r, g):assert 0 <= r and r <= self.nassert g(self.e)if r == 0:return 0r += self.sizefor i in range(self.log, 0, -1):self.push((r - 1) >> i)sm = self.ewhile 1:r -= 1while r > 1 and (r % 2):r >>= 1if not (g(self.op(self.d[r], sm))):while r < self.size:self.push(r)r = 2 * r + 1if g(self.op(self.d[r], sm)):sm = self.op(self.d[r], sm)r -= 1return r + 1 - self.sizesm = self.op(self.d[r], sm)if (r & -r) == r:breakreturn 0class segtree:n = 1size = 1log = 2d = [0]op = Nonee = 10**15def __init__(self, V, OP, E):self.n = len(V)self.op = OPself.e = Eself.log = (self.n - 1).bit_length()self.size = 1 << self.logself.d = [E for i in range(2 * self.size)]for i in range(self.n):self.d[self.size + i] = V[i]for i in range(self.size - 1, 0, -1):self.update(i)def set(self, p, x):assert 0 <= p and p < self.np += self.sizeself.d[p] = xfor i in range(1, self.log + 1):self.update(p >> i)def get(self, p):assert 0 <= p and p < self.nreturn self.d[p + self.size]def prod(self, l, r):assert 0 <= l and l <= r and r <= self.nsml = self.esmr = self.el += self.sizer += self.sizewhile l < r:if l & 1:sml = self.op(sml, self.d[l])l += 1if r & 1:smr = self.op(self.d[r - 1], smr)r -= 1l >>= 1r >>= 1return self.op(sml, smr)def all_prod(self):return self.d[1]def max_right(self, l, f):assert 0 <= l and l <= self.nassert f(self.e)if l == self.n:return self.nl += self.sizesm = self.ewhile 1:while l % 2 == 0:l >>= 1if not (f(self.op(sm, self.d[l]))):while l < self.size:l = 2 * lif f(self.op(sm, self.d[l])):sm = self.op(sm, self.d[l])l += 1return l - self.sizesm = self.op(sm, self.d[l])l += 1if (l & -l) == l:breakreturn self.ndef min_left(self, r, f):assert 0 <= r and r <= self.nassert f(self.e)if r == 0:return 0r += self.sizesm = self.ewhile 1:r -= 1while r > 1 and (r % 2):r >>= 1if not (f(self.op(self.d[r], sm))):while r < self.size:r = 2 * r + 1if f(self.op(self.d[r], sm)):sm = self.op(self.d[r], sm)r -= 1return r + 1 - self.sizesm = self.op(self.d[r], sm)if (r & -r) == r:breakreturn 0def update(self, k):self.d[k] = self.op(self.d[2 * k], self.d[2 * k + 1])def __str__(self):return str([self.get(i) for i in range(self.n)])def op(x, y):return x & ye = (1 << sigma) - 1def mapping(f, x):return ((x << f) & e) | (x >> (sigma - f))def composition(f, g):return (f + g) % sigmaid_ = 0S = input()T = input()diff = []for s, t in zip(S, T):diff.append(1 << ((ord(s) - ord(t)) % sigma))seg = lazy_segtree(diff, op, e, mapping, composition, id_)s_imos = segtree([0] * (len(S) + 1), lambda x, y: (x + y) % sigma, 0)t_imos = segtree([0] * (len(T) + 1), lambda x, y: (x + y) % sigma, 0)Q = int(input())for _ in range(Q):cmd, *query = list(map(int, input().split()))if cmd == 1:l, r, x = queryseg.apply(min(l - 1, len(S), len(T)), min(r, len(S), len(T)), x)s_imos.set(l - 1, s_imos.get(l - 1) + x)s_imos.set(r, s_imos.get(r) - x)elif cmd == 2:l, r, x = queryseg.apply(min(l - 1, len(S), len(T)), min(r, len(S), len(T)), -x % sigma)t_imos.set(l - 1, t_imos.get(l - 1) + x)t_imos.set(r, t_imos.get(r) - x)else:p = query[0]idx = seg.max_right(p - 1, lambda x: x & 1 == 1)if idx < min(len(S), len(T)):s = (s_imos.prod(0, idx + 1) + ord(S[idx]) - ord("a")) % sigmat = (t_imos.prod(0, idx + 1) + ord(T[idx]) - ord("a")) % sigmaif s > t:print("Greater")elif s < t:print("Lesser")else:if len(S) > len(T):print("Greater")elif len(S) < len(T):print("Lesser")else:print("Equals")