結果

問題 No.1014 competitive fighting
ユーザー mkawa2
提出日時 2020-05-16 15:42:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,420 ms / 2,000 ms
コード長 3,075 bytes
コンパイル時間 336 ms
コンパイル使用メモリ 82,460 KB
実行使用メモリ 107,484 KB
最終ジャッジ日時 2024-09-22 10:38:36
合計ジャッジ時間 35,798 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 51
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
from bisect import bisect_right
from heapq import *

sys.setrecursionlimit(10 ** 6)
int1 = lambda x: int(x) - 1
p2D = lambda x: print(*x, sep="\n")
def II(): return int(sys.stdin.readline())
def MI(): return map(int, sys.stdin.readline().split())
def LI(): return list(map(int, sys.stdin.readline().split()))
def LLI(rows_number): return [LI() for _ in range(rows_number)]
def SI(): return sys.stdin.readline()[:-1]

class SegtreeMax():
    def __init__(self, n):
        self.inf = 10 ** 16
        tree_width = 2
        while tree_width < n:
            tree_width *= 2
        self.tree_width = tree_width
        self.tree = [0] * (tree_width * 2 - 1)

    def update(self, i, a):
        seg_i = self.tree_width - 1 + i
        self.tree[seg_i] = a
        while seg_i != 0:
            seg_i = (seg_i - 1) // 2
            self.tree[seg_i] = max(self.tree[seg_i * 2 + 1], self.tree[seg_i * 2 + 2])

    def element(self, i):
        return self.tree[self.tree_width - 1 + i]

    def max(self, l, r, seg_i=0, segL=0, segR=-1):
        if segR == -1: segR = self.tree_width
        if r <= segL or segR <= l: return 0
        if l <= segL and segR <= r: return self.tree[seg_i]
        segM = (segL + segR) // 2
        ret0 = self.max(l, r, seg_i * 2 + 1, segL, segM)
        ret1 = self.max(l, r, seg_i * 2 + 2, segM, segR)
        return max(ret0, ret1)

def main():
    inf=10**15
    n=II()
    abci=[]
    for i in range(n):
        a,b,c=MI()
        abci.append((a,b,c,i))
    # aの小さいほうから見ていく
    # b-cが自分のaより小さければ、遷移先が出そろっているということなので
    # 遷移先の最大スコアにbを足して自分のスコアにする
    # スコアはセグメント木に入れておく
    # b-cがa以上であれば、b-c以下のaが出揃うまでhpに入れて保留する
    # 保留中はスコアをinfとしてセグメント木にいれておく
    # 保留中に自分への遷移があったときは、ループができるから
    abci.sort()
    ans=[0]*n
    st=SegtreeMax(n)
    hp=[]
    for j,(a,b,c,i) in enumerate(abci):
        # 保留中から出せるものを処理
        while hp and hp[0][0]<a:
            _,pb,pj=heappop(hp)
            if pj+1<j:mx=max(st.max(0,pj),st.max(pj+1,j))
            else:mx=st.max(0,pj)
            if mx!=inf:mx+=pb
            st.update(pj,mx)
            ans[abci[pj][3]]=mx
        # すぐに処理できるもの
        if b-c<a:
            k=bisect_right(abci,(b-c,inf,0,0))
            mx=st.max(0,k)
            if mx!=inf:mx+=b
            st.update(j,mx)
            ans[i]=mx
        # 保留するもの
        else:
            st.update(j,inf)
            heappush(hp,(b-c,b,j))
    # 最後まで保留されたものの処理
    while hp:
        _, pb, pj = heappop(hp)
        st.update(pj,0)
        mx=st.max(0,n)
        if mx!=inf:mx+=pb
        st.update(pj,mx)
        ans[abci[pj][3]]=mx
    # 答えの出力
    for x in ans:
        if x==inf:print("BAN")
        else:print(x)

main()
0