結果

問題 No.448 ゆきこーだーの雨と雪 (3)
ユーザー tktk_snsn
提出日時 2021-02-25 00:55:12
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,637 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 81,980 KB
実行使用メモリ 184,380 KB
最終ジャッジ日時 2024-09-25 01:42:59
合計ジャッジ時間 13,761 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 24 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

from operator import add
import bisect
from itertools import accumulate
import sys
input = sys.stdin.buffer.readline
sys.setrecursionlimit(10 ** 7)
inf = 10**18


class SparseTable:
    def __init__(self, N, arr, op, u):
        self.N = N
        self.depth = N.bit_length()
        self.op = op
        self.arr = tuple(arr)
        self.u = u
        self._build()

    def __call__(self, s, t):
        """
        [s, t)区間（0-indexed）での値を返す
        """
        if s == t:
            return self.u
        i = (t - s).bit_length() - 1
        L = self.table[i][s]
        R = self.table[i][t - (1 << i)]
        return self.op(L, R)

    def _build(self):
        self.table = []
        self.table.append(self.arr)
        N = self.N
        for i in range(self.depth - 1):
            shift = 1 << i
            N -= shift
            t = tuple(self.op(self.table[-1][j], self.table[-1][j+shift])
                      for j in range(N))
            self.table.append(t)



N, K = map(int, input().split())
T = [0] * N
D = [0] * N
for i in range(N):
    T[i], D[i] = map(int, input().split())
Dmax = SparseTable(N, D, max, 0)
Dacc = [0] + list(accumulate(D))
Dsum = lambda s, t: Dacc[t] - Dacc[s]

dp = [(inf, inf) for _ in range(N + 1)]
dp[0] = (0, 0)
for i in range(N):
    a, b = dp[i]

    #自分で処理
    na = max(a, D[i])
    nb = add(b, D[i])
    if dp[i + 1] > (na, nb):
        dp[i + 1] = (na, nb)

    #姉が処理
    to = bisect.bisect_left(T, T[i] + K)
    na = max(a, Dmax(i+1, to))
    nb = add(b, Dsum(i+1, to))
    if dp[to] > (na, nb):
        dp[to] = (na, nb)

print(*dp[N], sep="\n")
0