結果
問題 | No.2866 yuusaan's Knapsack |
ユーザー |
|
提出日時 | 2024-09-02 13:37:20 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,059 ms / 2,000 ms |
コード長 | 1,926 bytes |
コンパイル時間 | 175 ms |
コンパイル使用メモリ | 82,464 KB |
実行使用メモリ | 465,556 KB |
最終ジャッジ日時 | 2024-09-26 14:39:37 |
合計ジャッジ時間 | 20,650 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 26 |
ソースコード
import sys import math import bisect from heapq import heapify, heappop, heappush from collections import deque, defaultdict, Counter from functools import lru_cache from itertools import accumulate, combinations, permutations sys.setrecursionlimit(1000000) MOD = 10 ** 9 + 7 MOD99 = 998244353 input = lambda: sys.stdin.readline().strip() NI = lambda: int(input()) NMI = lambda: map(int, input().split()) NLI = lambda: list(NMI()) SI = lambda: input() SMI = lambda: input().split() SLI = lambda: list(SMI()) EI = lambda m: [NLI() for _ in range(m)] def main(): N, W = NMI() VW = EI(N) VW.sort(key=lambda x: x[1]) INF = 10**18 M = 30000 # dp[i][j]: i個みて、重さjのときのMax評価値と個数 dp = [[[-INF, 0] for _ in range(M)] for _ in range(N+1)] dp[0][0][0] = 0 dp[0][0][1] = 1 for i, (v, w) in enumerate(VW): # print(f"{i=} {v=} {w=}") for j in range(-10000, 10001): dv, dk = dp[i][j] # if dk > 0: # print(i, j, v, w, dv, dk) nv, nk = dp[i+1][j] if dv > nv: dp[i+1][j] = [dv, dk] elif dv == nv: dp[i+1][j][1] += dk dp[i+1][j][1] %= MOD99 if j + w > W: continue # if dk > 0: # print(f"# {i=} {j=} {v=} {w=} {j+w=} {dv=} {dv+v=} {dk=}") nv, nk = dp[i+1][j+w] if dv+v > nv: dp[i+1][j+w] = [dv+v, dk] elif dv+v == nv: dp[i+1][j+w][1] += dk dp[i+1][j+w][1] %= MOD99 # for i in range(N+1): # print(dp[i][0]) tv = -INF for j in range(M): tv = max(tv, dp[-1][j][0]) ans = 0 for j in range(M): v, k = dp[-1][j] if v == tv: # print(j, v, k) ans += k print(tv, ans % MOD99) if __name__ == "__main__": main()