結果

問題 No.1014 competitive fighting
コンテスト
ユーザー mkawa2
提出日時 2020-03-24 12:06:03
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 3,039 bytes
コンパイル時間 202 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 79,212 KB
最終ジャッジ日時 2024-12-31 14:35:17
合計ジャッジ時間 56,853 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 41 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

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)
            st.update(pj,0)
            mx=st.max(0,j)
            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