結果

問題 No.2617 容量3のナップザック
ユーザー TairitsuMeowTairitsuMeow
提出日時 2024-01-26 23:08:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 235 ms / 2,000 ms
コード長 1,774 bytes
コンパイル時間 2,001 ms
コンパイル使用メモリ 170,788 KB
実行使用メモリ 50,560 KB
最終ジャッジ日時 2024-09-28 08:59:10
合計ジャッジ時間 7,414 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff # 

// Problem: No.2617 容量3のナップザック
// Contest: yukicoder
// URL: https://yukicoder.me/problems/no/2617
// Memory Limit: 512 MB
// Time Limit: 2000 ms

#include<bits/stdc++.h>
#define debug(x) cerr<<(#x)<<" "<<(x)<<endl
typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
#define pii pair<ll,ll>
#define rep(i,a,b) for(ll i=(a);i<=(b);++i)
#define per(i,a,b) for(ll i=(a);i>=(b);--i)
using namespace std;
bool Mbe; 
ll read(){
	ll x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
void write(ll x){
	if(x<0)putchar('-'),x=-x;
	if(x>9)write(x/10);
	putchar(x%10+'0');
}
const ll N=2e6+9;
ll n,k,seed,A,B,m,v[N],f[N],w[N],b1[N],b2[N],b3[N],t1,t2,t3,p1,p2,c,sum,ans,sum1[N],sum2[N];
ll F(ll k,ll x){
	return sum1[min(t1,3*k-x*2)]+sum2[x];
}
ll solve(ll k){
	ll l=0,r=min(t2,k);
	while(r-l>20){
		ll lmid=l+(r-l)/3;
		ll rmid=r-(r-l)/3;
		if(F(k,lmid)>F(k,rmid))r=rmid;
		else l=lmid;
	}
	ll res=0;
	rep(i,l,r)res=max(res,F(k,i));
	return res;
}
bool Med;
int main(){
	cerr<<fabs(&Med-&Mbe)/1048576.0<<"MB\n";
	n=read(),k=read(),seed=read(),A=read(),B=read(),m=read();
	f[1]=seed;
	rep(i,2,2*n)f[i]=(A*f[i-1]+B)%m;
	rep(i,1,n){
		w[i]=f[i]%3+1;
		v[i]=f[i+n]*w[i];
		if(w[i]==1)b1[++t1]=v[i];
		else if(w[i]==2)b2[++t2]=v[i];
		else b3[++t3]=v[i];
	}
	sort(b1+1,b1+t1+1);
	sort(b2+1,b2+t2+1);
	sort(b3+1,b3+t3+1);
	reverse(b1+1,b1+t1+1);
	reverse(b2+1,b2+t2+1);
	reverse(b3+1,b3+t3+1);
	rep(i,1,min(t3,k))sum+=b3[i];
	rep(i,1,t1)sum1[i]=sum1[i-1]+b1[i];
	rep(i,1,t2)sum2[i]=sum2[i-1]+b2[i];
	ll tk=k-min(t3,k);
	ans=sum+solve(tk);
	per(i,min(k-1,t3-1),0){
		sum-=b3[i+1];
		tk++;
		ans=max(ans,sum+solve(tk));
	}
	write(ans);
	return 0;
}
0