結果
問題 |
No.2866 yuusaan's Knapsack
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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)