Submission #953811

#	Time	Username	Problem	Language	Result	Execution time	Memory
953811		PM1	Uplifting Excursion (BOI22_vault)	C++17	50 / 100	653 ms	3432 KiB

vault

#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define fr first
#define sc second
const int mxn=305;
ll m,a[mxn*2],l,sum,ans,num;
vector<pair<ll,ll>>v;
ll dp[2][mxn*mxn*2];
void add(ll tedad,ll x,ll z){
	tedad=(min(tedad,m*2));
	for(ll i=1;i<=tedad;i*=2){
		v.push_back({i*x,z*i});
		tedad-=i;
	}
	if(tedad>0)
		v.push_back({tedad*x,z*tedad});
}
void solve(){
	for(ll i=m;i<2*m+1;i++){
		ans+=a[i];
		num+=a[i]*(i-m);
		add(a[i],-(i-m),-1);
	}
	ll i=m-1;
	for(;i>=0;i--){
		if(num+a[i]*(i-m)>=l){
			ans+=a[i];
			num+=a[i]*(i-m);
			add(a[i],-(i-m),-1);
			continue;
		}
		ll x=l-num;
		x=x/(i-m);
		x=min(x,a[i]);
		num+=a[i]*x;
		ans+=x;
		add(x,-(i-m),-1);
		add(a[i]-x,i-m,1);
		i--;
		break;
	}
	for(;i>=0;i--){
		add(a[i],(i-m),1);
	}
}
void solve1(){
	for(ll i=m;i>=0;i--){
		ans+=a[i];
		num+=a[i]*(i-m);
		add(a[i],-(i-m),-1);
	}
	ll i=m+1;
	for(;i<m*2+1;i++){
		if(num+a[i]*(i-m)<=l){
			ans+=a[i];
			num+=a[i]*(i-m);
			add(a[i],-(i-m),-1);
			continue;
		}
		ll x=l-num;
		x=x/(i-m);
		assert(x>=0);
		x=min(x,a[i]);
		num+=(i-m)*x;
		ans+=x;
		add(x,-(i-m),-1);
		add(a[i]-x,i-m,1);
		i++;
		break;
	}
	for(;i<m*2+1;i++){
		add(a[i],(i-m),1);
	}
}
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>m>>l;
	for(ll i=0;i<m*2+1;i++){
		cin>>a[i];
		sum+=(i-m)*a[i];
	}
	if(sum<=l)
		solve();
	else
		solve1();

	for(int i=0;i<=2*m*m;i++)dp[1][i]=-1e18;
	dp[1][m*m]=0;
	for(int i=0;i<v.size();i++){
		ll x=v[i].fr,y=v[i].sc;
		int z=i&1;
		for(ll j=0;j<=2*m*m;j++){
			if(j-x>=0 && j-x<=2*m*m && dp[z^1][j-x]!=-1e18)
				dp[z][j]=max(dp[z^1][j],dp[z^1][j-x]+y);
			else
				dp[z][j]=dp[z^1][j];
		}
		//cout<<dp[z][928]<<" ";
	}
	int x=v.size()&1;
	x^=1;
	l-=num;
	if( l>m*m || l<-m*m ||dp[x][l+m*m]==-1e18 || ans+dp[x][l+m*m]<0){
		cout<<"impossible";
		return 0;
	}
	cout<<ans+dp[x][l+m*m];
	return 0;
}

Compilation message (stderr)

vault.cpp: In function 'int main()':
vault.cpp:92:15: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   92 |  for(int i=0;i<v.size();i++){
      |              ~^~~~~~~~~

• Subtask 1: 0 / 5
• Subtask 2: 0 / 15
• Subtask 3-1: 10 / 10
• Subtask 3-2: 10 / 10
• Subtask 4-1: 10 / 10
• Subtask 4-2: 0 / 10
• Subtask 5-1: 10 / 10
• Subtask 5-2: 0 / 10
• Subtask 6-1: 10 / 10
• Subtask 6-2: 0 / 10