結果
問題 | No.1826 Fruits Collecting |
ユーザー |
|
提出日時 | 2022-01-28 23:26:57 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 971 ms / 2,000 ms |
コード長 | 1,855 bytes |
コンパイル時間 | 595 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 153,856 KB |
最終ジャッジ日時 | 2024-12-30 11:52:25 |
合計ジャッジ時間 | 23,596 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 43 |
ソースコード
class SegmentTree: def __init__(self, init_val, segfunc, ide_ele): n = len(init_val) self.segfunc = segfunc self.ide_ele = ide_ele self.num = 1 << (n - 1).bit_length() self.tree = [ide_ele] * 2 * self.num self.size = n for i in range(n): self.tree[self.num + i] = init_val[i] for i in range(self.num - 1, 0, -1): self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1]) def update(self, k, x): k += self.num self.tree[k] = self.segfunc(self.tree[k],x) while k > 1: k >>= 1 self.tree[k] = self.segfunc(self.tree[2*k], self.tree[2*k+1]) def query(self, l, r): if r==self.size: r = self.num res = self.ide_ele l += self.num r += self.num right = [] while l < r: if l & 1: res = self.segfunc(res, self.tree[l]) l += 1 if r & 1: right.append(self.tree[r-1]) l >>= 1 r >>= 1 for e in right[::-1]: res = self.segfunc(res,e) return res import sys,random,bisect from collections import deque,defaultdict from heapq import heapify,heappop,heappush from itertools import permutations from math import log,gcd input = lambda :sys.stdin.readline().rstrip() mi = lambda :map(int,input().split()) li = lambda :list(mi()) N = int(input()) item = [tuple(mi()) for i in range(N)] item.sort(key=lambda x:x[0]-x[1]) val = set([x[0]+x[1] for x in item]+[0]) comp = {e:i for i,e in enumerate(sorted(val))} n = len(comp) dp = SegmentTree([-10**17]*n,max,-10**17) dp.update(comp[0],0) for t,x,v in item: if t-x < 0: continue tmp = dp.query(0,comp[t+x]+1) + v dp.update(comp[t+x],tmp) print(dp.query(0,n))