結果

問題 No.2617 容量3のナップザック
ユーザー とりゐ
提出日時 2024-01-26 22:05:51
言語 PyPy3
(7.3.15)
結果
RE  
実行時間 -
コード長 814 bytes
コンパイル時間 272 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 190,720 KB
最終ジャッジ日時 2024-09-28 08:12:45
合計ジャッジ時間 4,419 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 8 RE * 2 TLE * 1 -- * 29
権限があれば一括ダウンロードができます

ソースコード

diff # 

N,K,seed,a,b,m=map(int,input().split())
w=[0]*N
f=seed
for i in range(N):
  w[i]=f%3+1
  f=(a*f+b)%m

v=[0]*N
for i in range(N):
  v[i]=f*w[i]
  f=(a*f+b)%m

w1=[0]*3*K
w2=[0]*K
w3=[0]*K
for i in range(N):
  if w[i]==1:
    w1.append(v[i])
  if w[i]==2:
    w2.append(v[i])
  if w[i]==3:
    w3.append(v[i])

w1.sort(reverse=True)
w2.sort(reverse=True)
w3.sort(reverse=True)

ans=0
res=[0]
idx1=0
idx2=0
tmp=0
for i in range(1,K+1):
  tmp+=w1[idx1]+w1[idx1+1]+w1[idx1+2]
  idx1+=3
  while idx2<i and w2[idx2]>=w1[idx1-1]+w1[idx1-2]:
    tmp+=w2[idx2]-w1[idx1-1]-w1[idx1-2]
    idx2+=1
    idx1-=2
  while idx2>0 and w1[idx1]+w1[idx1+1]>=w2[idx2-1]:
    tmp+=w1[idx1]+w1[idx1+1]-w2[idx2-1]
    idx2-=1
    idx1+=2
  res.append(tmp)

cum3=0
for i in range(K+1):
  ans=max(ans,res[K-i]+cum3)
  cum3+=w3[i]

print(ans)
0