結果
問題 | No.1826 Fruits Collecting |
ユーザー |
|
提出日時 | 2022-01-28 22:55:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,181 ms / 2,000 ms |
コード長 | 2,002 bytes |
コンパイル時間 | 765 ms |
コンパイル使用メモリ | 81,920 KB |
実行使用メモリ | 196,320 KB |
最終ジャッジ日時 | 2024-12-30 10:17:26 |
合計ジャッジ時間 | 26,635 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 43 |
ソースコード
class SegmentTree:def __init__(self, init_val, segfunc, ide_ele):n = len(init_val)self.segfunc = segfuncself.ide_ele = ide_eleself.num = 1 << (n - 1).bit_length()self.tree = [ide_ele] * 2 * self.numself.size = nfor 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.numself.tree[k] = self.segfunc(self.tree[k],x)while k > 1:k >>= 1self.tree[k] = self.segfunc(self.tree[2*k], self.tree[2*k+1])def query(self, l, r):if r==self.size:r = self.numres = self.ide_elel += self.numr += self.numright = []while l < r:if l & 1:res = self.segfunc(res, self.tree[l])l += 1if r & 1:right.append(self.tree[r-1])l >>= 1r >>= 1for e in right[::-1]:res = self.segfunc(res,e)return resimport sys,random,bisectfrom collections import deque,defaultdictfrom heapq import heapify,heappop,heappushfrom itertools import permutationsfrom math import log,gcdinput = 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)]dic = {}for t,x,v in item:if (t,x) not in dic:dic[t,x] = 0dic[t,x] += vitem = [(t,x,dic[t,x]) for (t,x) in dic]N = len(item)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:continuetmp = dp.query(0,comp[t+x]+1) + vdp.update(comp[t+x],tmp)print(dp.query(0,n))