結果

問題 No.2866 yuusaan's Knapsack
ユーザー Mao-beta
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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()
0