結果
問題 | No.1234 典型RMQ |
ユーザー | Kiri8128 |
提出日時 | 2020-09-19 03:04:49 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 495 ms / 2,000 ms |
コード長 | 7,850 bytes |
コンパイル時間 | 331 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 104,704 KB |
最終ジャッジ日時 | 2024-11-09 02:09:32 |
合計ジャッジ時間 | 12,551 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 47 ms
54,656 KB |
testcase_01 | AC | 47 ms
54,144 KB |
testcase_02 | AC | 46 ms
54,016 KB |
testcase_03 | AC | 47 ms
54,528 KB |
testcase_04 | AC | 48 ms
54,272 KB |
testcase_05 | AC | 48 ms
54,016 KB |
testcase_06 | AC | 479 ms
100,096 KB |
testcase_07 | AC | 423 ms
80,860 KB |
testcase_08 | AC | 488 ms
104,192 KB |
testcase_09 | AC | 456 ms
89,472 KB |
testcase_10 | AC | 487 ms
102,272 KB |
testcase_11 | AC | 472 ms
99,968 KB |
testcase_12 | AC | 451 ms
88,448 KB |
testcase_13 | AC | 417 ms
80,128 KB |
testcase_14 | AC | 461 ms
88,960 KB |
testcase_15 | AC | 470 ms
87,552 KB |
testcase_16 | AC | 495 ms
102,528 KB |
testcase_17 | AC | 463 ms
88,960 KB |
testcase_18 | AC | 392 ms
79,408 KB |
testcase_19 | AC | 493 ms
104,064 KB |
testcase_20 | AC | 323 ms
102,784 KB |
testcase_21 | AC | 475 ms
99,712 KB |
testcase_22 | AC | 444 ms
104,064 KB |
testcase_23 | AC | 427 ms
104,704 KB |
testcase_24 | AC | 436 ms
104,704 KB |
testcase_25 | AC | 428 ms
104,704 KB |
testcase_26 | AC | 440 ms
104,704 KB |
testcase_27 | AC | 47 ms
54,400 KB |
testcase_28 | AC | 48 ms
54,528 KB |
testcase_29 | AC | 48 ms
54,400 KB |
ソースコード
import sys input = lambda: sys.stdin.readline().rstrip() class SegmentTree(): def __init__(self, init, unitX, unitA, f, g, h): self.f = f # (X, X) -> X self.g = g # (X, A, size) -> X self.h = h # (A, A) -> A self.unitX = unitX self.unitA = unitA self.f = f if type(init) == int: self.n = init self.n = 1 << (self.n - 1).bit_length() self.X = [unitX] * (self.n * 2) self.size = [1] * (self.n * 2) else: self.n = len(init) self.n = 1 << (self.n - 1).bit_length() self.X = [unitX] * self.n + init + [unitX] * (self.n - len(init)) self.size = [0] * self.n + [1] * len(init) + [0] * (self.n - len(init)) for i in range(self.n-1, 0, -1): self.X[i] = self.f(self.X[i*2], self.X[i*2|1]) for i in range(self.n - 1, 0, -1): self.size[i] = self.size[i*2] + self.size[i*2|1] self.A = [unitA] * (self.n * 2) def update(self, i, x): i += self.n self.propagate_above(i) self.X[i] = x self.A[i] = unitA self.calc_above(i) def calc(self, i): return self.g(self.X[i], self.A[i], self.size[i]) def calc_above(self, i): i >>= 1 while i: self.X[i] = self.f(self.calc(i*2), self.calc(i*2|1)) i >>= 1 def propagate(self, i): self.X[i] = self.g(self.X[i], self.A[i], self.size[i]) self.A[i*2] = self.h(self.A[i*2], self.A[i]) self.A[i*2|1] = self.h(self.A[i*2|1], self.A[i]) self.A[i] = self.unitA def propagate_above(self, i): H = i.bit_length() for h in range(H, 0, -1): self.propagate(i >> h) def propagate_all(self): for i in range(1, self.n): self.propagate(i) def getrange(self, l, r): l += self.n r += self.n l0, r0 = l // (l & -l), r // (r & -r) - 1 self.propagate_above(l0) self.propagate_above(r0) al = self.unitX ar = self.unitX while l < r: if l & 1: al = self.f(al, self.calc(l)) l += 1 if r & 1: r -= 1 ar = self.f(self.calc(r), ar) l >>= 1 r >>= 1 return self.f(al, ar) def getvalue(self, i): i += self.n self.propagate_above(i) return self.calc(i) def operate_range(self, l, r, a): l += self.n r += self.n l0, r0 = l // (l & -l), r // (r & -r) - 1 self.propagate_above(l0) self.propagate_above(r0) while l < r: if l & 1: self.A[l] = self.h(self.A[l], a) l += 1 if r & 1: r -= 1 self.A[r] = self.h(self.A[r], a) l >>= 1 r >>= 1 self.calc_above(l0) self.calc_above(r0) def check(self, randX, randA, maxs, rep): from random import randrange for _ in range(rep): x = randX() y = randX() z = randX() a = randA() b = randA() c = randA() s = randrange(1, maxs + 1) t = randrange(1, maxs + 1) err = 0 if not f(x, unitX) == f(unitX, x) == x: err = 1 print("!!!!! unitX Error !!!!!") print("unitX =", unitX) print("x =", x) print("f(x, unitX) =", f(x, unitX)) print("f(unitX, x) =", f(unitX, x)) if not h(a, unitA) == h(unitA, a) == a: err = 1 print("!!!!! unitA Error !!!!!") print("unitA =", unitA) print("a =", a) print("f(a, unitA) =", f(a, unitA)) print("f(unitA, a) =", f(unitA, a)) if not f(f(x, y), z) == f(x, f(y, z)): err = 1 print("!!!!! Associativity Error X !!!!!") print("x, y, z, f(x, y), f(y, x) =", x, y, z, f(x, y), f(y, x)) print("f(f(x, y), z) =", f(f(x, y), z)) print("f(x, f(y, z)) =", f(x, f(y, z))) if not h(h(a, b), c) == h(a, h(b, c)): err = 1 print("!!!!! Associativity Error A !!!!!") print("a, b, c, h(a, b), h(b, c) =", a, b, c, h(a, b), h(b, c)) print("h(h(a, b), c) =", h(h(a, b), c)) print("h(a, h(b, c)) =", h(a, h(b, c))) if not g(x, unitA, s) == x: err = 1 print("!!!!! Identity Error !!!!!") print("unitA, x, s =", unitA, x, s) print("g(x, unitA, s) =", g(x, unitA, s)) if not g(g(x, a, s), b, s) == g(x, h(a, b), s): err = 1 print("!!!!! Act Error 1 !!!!!") print("x, a, b, s, g(x, a, s), h(a, b) =", x, a, b, s, g(x, a, s), h(a, b)) print("g(g(x, a, s), b, s) =", g(g(x, a, s), b, s)) print("g(x, h(a, b), s) =", g(x, h(a, b), s)) if not g(f(x, y), a, s + t) == f(g(x, a, s), g(y, a, t)): err = 1 print("!!!!! Act Error 2 !!!!!") print("x, y, a, s, t, f(x, y), g(x, a, s), g(y, a, t) =", x, y, a, s, t, f(x, y), g(x, a, s), g(y, a, t)) print("g(f(x, y), a, s + t) =", g(f(x, y), a, s + t)) print("f(g(x, a, s), g(y, a, t)) =", f(g(x, a, s), g(y, a, t))) if err: break assert f(x, unitX) == f(unitX, x) == x assert h(a, unitA) == h(unitA, a) == a assert f(f(x, y), z) == f(x, f(y, z)) assert h(h(a, b), c) == h(a, h(b, c)) assert g(x, unitA, s) == x assert g(g(x, a, s), b, s) == g(x, h(a, b), s) assert g(f(x, y), a, s + t) == f(g(x, a, s), g(y, a, t)) else: pass print("Monoid Check OK!") def debug1(self): print("self.n =", self.n) deX = [] deA = [] deS = [] a, b = self.n, self.n * 2 while b: deX.append(self.X[a:b]) deA.append(self.A[a:b]) deS.append(self.size[a:b]) a, b = a//2, a print("--- debug ---") for d in deX[::-1]: print(d) print("--- ---") for d in deA[::-1]: print(d) print("--- ---") for d in deS[::-1]: print(d) print("--- ---") def debug(self, k = 10): print("--- debug ---") print("point") for i in range(min(self.n - 1, k)): print(i, self.getvalue(i)) print("prod") for i in range(min(self.n, k)): print(i, self.getrange(0, i)) print("--- ---") from operator import add # f = lambda x, y: (x[0] + y[0], x[1] + y[1]) # g = lambda x, a, s: (s, 0) if a == "A" else (0, s) if a == "B" else (x) # h = lambda a, b: b if len(b) else a f = min g = lambda x, a, s: x + a h = add unitX = 1 << 61 unitA = 0 N = int(input()) A = [int(a) for a in input().split()] st = SegmentTree(A, unitX, unitA, f, g, h) if False: from random import randrange randX = lambda: randrange(-10, 11) randA = lambda: randrange(-10, 11) maxs = 10 rep = 1000 st.check(randX, randA, maxs, rep) Q = int(input()) for _ in range(Q): k, l, r, c = map(int, input().split()) l -= 1 if k == 1: st.operate_range(l, r, c) else: print(st.getrange(l, r))