結果
問題 | No.2974 関数の芽 |
ユーザー | hiro1729 |
提出日時 | 2023-12-16 18:34:31 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,884 bytes |
コンパイル時間 | 368 ms |
コンパイル使用メモリ | 82,004 KB |
実行使用メモリ | 175,416 KB |
最終ジャッジ日時 | 2024-09-22 19:29:56 |
合計ジャッジ時間 | 8,834 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 45 ms
56,012 KB |
testcase_01 | AC | 44 ms
54,760 KB |
testcase_02 | AC | 43 ms
55,268 KB |
testcase_03 | AC | 43 ms
54,888 KB |
testcase_04 | AC | 42 ms
54,588 KB |
testcase_05 | AC | 43 ms
54,604 KB |
testcase_06 | AC | 44 ms
55,528 KB |
testcase_07 | AC | 44 ms
54,628 KB |
testcase_08 | AC | 45 ms
55,392 KB |
testcase_09 | WA | - |
testcase_10 | AC | 43 ms
55,624 KB |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | AC | 224 ms
80,420 KB |
testcase_14 | AC | 364 ms
83,924 KB |
testcase_15 | AC | 647 ms
174,876 KB |
testcase_16 | AC | 483 ms
174,736 KB |
testcase_17 | AC | 662 ms
174,740 KB |
testcase_18 | WA | - |
testcase_19 | WA | - |
testcase_20 | WA | - |
testcase_21 | WA | - |
testcase_22 | WA | - |
testcase_23 | AC | 700 ms
175,008 KB |
ソースコード
from collections import defaultdict from bisect import bisect_left as bsl class RangeAddSegTree: def __init__(self, n): self.l = 1 << (n - 1).bit_length() self.t = ([0] * self.l + [0] * n + [0] * (self.l - n)) def add(self, l, r, x): l += self.l r += self.l while l < r: if l & 1: self.t[l] = x + self.t[l] l += 1 if r & 1: self.t[r - 1] = x + self.t[r - 1] r -= 1 l >>= 1 r >>= 1 def get(self, x): x += self.l res = 0 while x > 0: res = res + self.t[x] x >>= 1 return res Q = int(input()) K = [0] * Q L = [0] * Q M = [0] * Q N = [0] * Q X = [0] * Q Xs = set() for i in range(Q): K[i], L[i], M[i], N[i], X[i] = map(int, input().split()) Xs.add(X[i]) Xs = list(Xs) lx = len(Xs) Xt = defaultdict(int) for i in range(lx): Xt[Xs[i]] = i FL = [RangeAddSegTree(lx) for _ in range(2)] FR = [RangeAddSegTree(lx) for _ in range(2)] GL = [RangeAddSegTree(lx) for _ in range(2)] GR = [RangeAddSegTree(lx) for _ in range(2)] for k, l, m, n, x in zip(K, L, M, N, X): if k == 0: FL[0].add(0, lx, max(0, l)) FR[0].add(0, lx, max(0, l)) else: x1 = -l // k + 1 x2 = (-l + k - 1) // k b1 = bsl(Xs, x1) b2 = bsl(Xs, x2) if k > 0: FL[0].add(b1, lx, l) FL[1].add(b1, lx, k) FR[0].add(b2, lx, l) FR[1].add(b2, lx, k) else: FL[0].add(0, b1, l) FL[1].add(0, b1, k) FR[0].add(0, b2, l) FR[1].add(0, b2, k) if m == 0: GL[0].add(0, lx, max(0, n)) GR[0].add(0, lx, max(0, n)) else: x1 = -n // m + 1 x2 = (-n + m - 1) // m b1 = bsl(Xs, x1) b2 = bsl(Xs, x2) if m > 0: GL[0].add(b1, lx, n) GL[1].add(b1, lx, m) GR[0].add(b2, lx, n) GR[1].add(b2, lx, m) else: GL[0].add(0, b1, n) GL[1].add(0, b1, m) GR[0].add(0, b2, n) GR[1].add(0, b2, m) p = Xt[x] print("Yes" if FL[0].get(p) == GL[0].get(p) and FL[1].get(p) == GL[1].get(p) and FR[1].get(p) == GR[1].get(p) else "No")