結果

問題 No.2617 容量3のナップザック
ユーザー とりゐとりゐ
提出日時 2024-01-26 22:07:55
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 446 ms / 2,000 ms
コード長 892 bytes
コンパイル時間 339 ms
コンパイル使用メモリ 82,212 KB
実行使用メモリ 255,452 KB
最終ジャッジ日時 2024-09-28 08:14:20
合計ジャッジ時間 10,685 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

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=[]
w2=[]
w3=[]
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)

w1+=[0]*max(0,(3*N+10-len(w1)))
w2+=[0]*max(0,N+10-len(w2))
w3+=[0]*max(0,N+10-len(w3))


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