結果

問題 No.2866 yuusaan's Knapsack
コンテスト
ユーザー katonyonko
提出日時 2024-09-10 18:29:08
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,377 ms / 2,000 ms
コード長 2,227 bytes
コンパイル時間 290 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 434,816 KB
最終ジャッジ日時 2024-09-26 14:40:07
合計ジャッジ時間 22,478 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

import io
import sys
from collections import defaultdict, deque, Counter
from itertools import permutations, combinations, accumulate
from heapq import heappush, heappop
from bisect import bisect_right, bisect_left
from math import gcd
import math

_INPUT = """\
6
5 10
4 3
2 1
5 5
4 4
3 3
2 0
-100 -100
500 100
6 -2
6 5
1 -6
6 4
6 6
-3 -5
7 -2
"""

def input():
  return sys.stdin.readline()[:-1]

inf=10**20
mod=998244353
def solve(test):
  N,W=map(int, input().split())
  bag=sorted([list(map(int, input().split())) for _ in range(N)],key=lambda x:x[1])
  def idx(i,j):
    return i*(20000+1)+j
  dp=[[-inf,0] for _ in range((N+1)*(20000+1))]
  dp[idx(0,10000)]=[0,1]
  # print(bag)
  for i in range(N):
    v,w=bag[i]
    for j in range(20000+1):
      if dp[idx(i+1,j)][0]<dp[idx(i,j)][0]:
        dp[idx(i+1,j)][1]=dp[idx(i,j)][1]
      elif dp[idx(i+1,j)][0]==dp[idx(i,j)][0]:
        dp[idx(i+1,j)][1]+=dp[idx(i,j)][1]
        dp[idx(i+1,j)][1]%=mod
      dp[idx(i+1,j)][0]=max(dp[idx(i+1,j)][0],dp[idx(i,j)][0])
      if 0<=j+w<=20000:
        if dp[idx(i+1,j+w)][0]<dp[idx(i,j)][0]+v:
          dp[idx(i+1,j+w)][1]=dp[idx(i,j)][1]
        elif dp[idx(i+1,j+w)][0]==dp[idx(i,j)][0]+v:
          dp[idx(i+1,j+w)][1]+=dp[idx(i,j)][1]
          dp[idx(i+1,j+w)][1]%=mod
        dp[idx(i+1,j+w)][0]=max(dp[idx(i+1,j+w)][0],dp[idx(i,j)][0]+v)
  m=max([dp[idx(N,j)][0] for j in range((W+10000+1))])
  print(m,sum([dp[idx(N,j)][1] for j in range((W+10000+1)) if dp[idx(N,j)][0]==m])%mod)

def random_input():
  from random import randint,shuffle
  N=randint(1,10)
  M=randint(1,N)
  A=list(range(1,M+1))+[randint(1,M) for _ in range(N-M)]
  shuffle(A)
  return (" ".join(map(str, [N,M]))+"\n"+" ".join(map(str, A))+"\n")*3

def simple_solve():
  return []

def main(test):
  if test==0:
    solve(0)
  elif test==1:
    sys.stdin = io.StringIO(_INPUT)
    case_no=int(input())
    for _ in range(case_no):
      solve(0)
  else:
    for i in range(1000):
      sys.stdin = io.StringIO(random_input())
      x=solve(1)
      y=simple_solve()
      if x!=y:
        print(i,x,y)
        print(*[line for line in sys.stdin],sep='')
        break

#0:提出用、1:与えられたテスト用、2:ストレステスト用
main(0)
0