結果

問題 No.951 【本日限定】1枚頼むともう1枚無料!
ユーザー maspy
提出日時 2020-01-06 10:19:53
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 651 ms / 2,000 ms
コード長 849 bytes
コンパイル時間 145 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 44,544 KB
最終ジャッジ日時 2024-11-22 23:50:27
合計ジャッジ時間 34,534 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 52
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

# 高い商品から順に見る。購入対象を固定すると、無料の権利は即時消費するとしてよい。

from operator import itemgetter
import numpy as np

N,K = map(int,readline().split())
m = map(int,read().split())
PD = sorted(zip(m,m), key = itemgetter(0), reverse=True)

# 購入合計金額 -> opt 価値
INF = 10 ** 18
dp_0 = np.full(K+1,-INF,np.int64); dp_0[0] = 0
dp_1 = np.full(K+1,-INF,np.int64) # 次は無料でもらえる

for p,d in PD:
    prev_0 = dp_0.copy(); prev_1 = dp_1.copy()
    # 無料で買う場合
    np.maximum(prev_1 + d, dp_0, out=dp_0)
    # 有料で買う場合
    np.maximum(prev_0[0:K+1-p] + d, dp_1[p:], out = dp_1[p:])

answer = max(dp_0.max(), dp_1.max())
print(answer)
0