結果
問題 | No.1826 Fruits Collecting |
ユーザー |
![]() |
提出日時 | 2022-01-28 22:48:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,241 ms / 2,000 ms |
コード長 | 2,942 bytes |
コンパイル時間 | 550 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 173,732 KB |
最終ジャッジ日時 | 2024-12-30 09:34:41 |
合計ジャッジ時間 | 25,736 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 43 |
ソースコード
mod = 1000000007 eps = 10**-9 def main(): import sys input = sys.stdin.buffer.readline class SegmentTree: def __init__(self, A, initialize=True, segfunc=max, ident=-10 ** 18): self.N = len(A) self.LV = (self.N - 1).bit_length() self.N0 = 1 << self.LV self.segfunc = segfunc self.ident = ident if initialize: self.data = [self.ident] * self.N0 + A + [self.ident] * (self.N0 - self.N) for i in range(self.N0 - 1, 0, -1): self.data[i] = segfunc(self.data[i * 2], self.data[i * 2 + 1]) else: self.data = [self.ident] * (self.N0 * 2) def update(self, i, x): i += self.N0 - 1 self.data[i] = x for _ in range(self.LV): i >>= 1 self.data[i] = self.segfunc(self.data[i * 2], self.data[i * 2 + 1]) def get(self, i): return self.data[i + self.N0 - 1] # open interval [l, r) def query(self, l, r): l += self.N0 - 1 r += self.N0 - 1 ret_l = self.ident ret_r = self.ident while l < r: if l & 1: ret_l = self.segfunc(ret_l, self.data[l]) l += 1 if r & 1: ret_r = self.segfunc(self.data[r - 1], ret_r) r -= 1 l >>= 1 r >>= 1 return self.segfunc(ret_l, ret_r) # return smallest i(l <= i < r) s.t. check(A[i]) == True def binsearch(self, l, r, check): if not check(self.query(l, r)): return r l += self.N0 - 1 val = self.ident while True: if check(self.segfunc(val, self.data[l])): break if l & 1: val = self.segfunc(val, self.data[l]) l += 1 l >>= 1 while l < self.N0: newval = self.segfunc(val, self.data[l * 2]) if not check(newval): val = newval l = (l << 1) + 1 else: l <<= 1 return l - self.N0 + 1 N = int(input()) XYV = [] Y = {0} for _ in range(N): t, x, v = map(int, input().split()) xx = t - x yy = t + x XYV.append((xx, yy, v)) Y.add(yy) XYV.sort(key=lambda z: z[1]) XYV.sort(key=lambda z: z[0]) Y = sorted(list(Y)) idx = {y: i + 1 for i, y in enumerate(Y)} L = len(Y) ST = SegmentTree([-10 ** 17] * L) ST.update(idx[0], 0) for x, y, v in XYV: if x < 0: continue i = idx[y] val = ST.query(1, i + 1) ST.update(i, val + v) print(ST.query(1, L+1)) if __name__ == '__main__': main()