Submission #428581

#TimeUsernameProblemLanguageResultExecution timeMemory
428581kshitij_sodaniShortcut (IOI16_shortcut)C++14
100 / 100
2000 ms139400 KiB
#include <bits/stdc++.h>
using namespace std;
typedef long long llo;
#define a first
#define b second
#define pb push_back


#include "shortcut.h"
llo pre[1000001];
int n;
llo it[1000001];
llo ind[1000001];
llo ind2[1000001];
llo ind3[1000001];
llo ind4[1000001];
llo ind5[1000001];
llo tree[4*1000001][4];

llo val[1000001];
llo val2[1000001];

llo ss;
llo ma[2][2];
vector<pair<llo,llo>> tt;
vector<pair<llo,llo>> tt2;

void update(llo no,llo l,llo r,llo i,llo val,llo ee){
	if(l==r){
		tree[no][ee]=val;
	}
	else{
		llo mid=(l+r)/2;
		if(i<=mid){
			update(no*2+1,l,mid,i,val,ee);
		}
		else{
			update(no*2+2,mid+1,r,i,val,ee);
		}
		tree[no][ee]=max(tree[no*2+1][ee],tree[no*2+2][ee]);
	}
}
llo query(llo no,llo l,llo r,llo aa,llo bb,llo ee){
	if(r<aa or l>bb){
		return -(llo)1e18;
	}
	if(aa<=l and r<=bb){
		return tree[no][ee];
	}
	llo mid=(l+r)/2;
	return max(query(no*2+1,l,mid,aa,bb,ee),query(no*2+2,mid+1,r,aa,bb,ee));
}
bool check(llo mid){

	llo ma=-1e18;
	llo ma2=-1e18;
	llo ma3=-1e18;
	llo ma4=-1e18;
	//llo st=1;
	/*for(int i=0;i<4*n;i++){
		for(int j=0;j<2;j++){
			tree[i][j]=-1e18;
		}
	}*/
	/*for(int j=0;j<n;j++){
		llo xx=-1;
		for(int k=16;k>=0;k--){
			if(xx+(1<<k)<n){
				if(tt[xx+(1<<k)].a+pre[j]+it[j]>mid){
					xx+=(1<<k);
				}
			}
		}
		if(xx>=0){
			llo yy=query(0,0,n-1,0,xx,0);
			llo zz=query(0,0,n-1,0,xx,1);
			ma=max(ma,yy+pre[j]+it[j]);
			ma2=max(ma2,yy-pre[j]+it[j]);
			ma3=max(ma3,zz-pre[j]+it[j]);
			ma4=max(ma4,zz+pre[j]+it[j]);
			//st=0;
		}


		update(0,0,n-1,ind5[j],pre[j]+it[j],0);
		update(0,0,n-1,ind5[j],-pre[j]+it[j],1);
	}*/
	pair<llo,llo> zz2={-1,-1};
	for(llo i=0;i<n;i++){
		zz2=max(zz2,{it[i],i});
	}
	llo cur=-1;
	for(auto ii:tt2){
		//cur=-1;
		while(cur+1<n){
			if(tt[cur+1].a+ii.a>mid){
				cur++;
			}
			else{
				break;
			}
		}
		if(ii.b==zz2.b){
		}
		else{
			/*for(auto jj:tt){
				if(jj.a+ii.a>mid){
					int i=jj.b;
					int j=ii.b;
					ma=max(ma,pre[i]+pre[j]+it[i]+it[j]);
				ma2=max(ma2,pre[i]-pre[j]+it[i]+it[j]);
				ma3=max(ma3,-pre[i]-pre[j]+it[i]+it[j]);
				ma4=max(ma4,-pre[i]+pre[j]+it[i]+it[j]);


				}
			}*/
			if(cur>=0){
				int j=ii.b;
				llo yy=val[cur];
				llo zz=val2[cur];
				ma=max(ma,yy+pre[j]+it[j]);
				ma2=max(ma2,yy-pre[j]+it[j]);
				ma3=max(ma3,zz-pre[j]+it[j]);
				ma4=max(ma4,zz+pre[j]+it[j]);
			}
		}
	}
	/*for(int i=0;i<n;i++){
		if(i==zz2.b){
			continue;
		}
		for(int j=0;j<n;j++){

			if(pre[j]-pre[i]+it[i]+it[j]>mid){
				//st=0;
				ma=max(ma,pre[i]+pre[j]+it[i]+it[j]);
				ma2=max(ma2,pre[i]-pre[j]+it[i]+it[j]);
				ma3=max(ma3,-pre[i]-pre[j]+it[i]+it[j]);
				ma4=max(ma4,-pre[i]+pre[j]+it[i]+it[j]);
			}
		}
	}*/
	for(int i=zz2.b;i<=zz2.b;i++){
		for(int j=zz2.b+1;j<n;j++){
			if(j==i){
				continue;
			}
			if(pre[j]-pre[i]+it[i]+it[j]>mid){
				//st=0;
				ma=max(ma,pre[i]+pre[j]+it[i]+it[j]);
				ma2=max(ma2,pre[i]-pre[j]+it[i]+it[j]);
				ma3=max(ma3,-pre[i]-pre[j]+it[i]+it[j]);
				ma4=max(ma4,-pre[i]+pre[j]+it[i]+it[j]);
			}
		}
	}
	for(int j=zz2.b;j<=zz2.b;j++){
		for(int i=0;i<zz2.b;i++){
			if(i==j){
				continue;
			}
			if(pre[j]-pre[i]+it[i]+it[j]>mid){
				//st=0;
				ma=max(ma,pre[i]+pre[j]+it[i]+it[j]);
				ma2=max(ma2,pre[i]-pre[j]+it[i]+it[j]);
				ma3=max(ma3,-pre[i]-pre[j]+it[i]+it[j]);
				ma4=max(ma4,-pre[i]+pre[j]+it[i]+it[j]);
			}
		}
	}
	/*if(st==1){
		return true;
	}*/
	//return false;
	ma=mid-ma-ss;
	ma=-ma;
	ma2=mid-ma2-ss;
	ma2=-ma2;
	ma3=mid-ma3-ss;
	ma4=mid-ma4-ss;
	//-pre[i]-pre[j]<=ma
	//pre[i]+pre[j]>=ma
	//pre[i]+pre[j]<=ma3

	//-pre[i]+pre[j]<=ma2
	//pre[i]-pre[j]>=ma2
	//pre[i]-pre[j]<=ma4

/*	llo ind=0;
	llo ind3=0;
	llo ind2=0;
	llo ind4=0;*/
	if(pre[n-1]+pre[n-2]<ma){
		return false;
	}
	if(ma>ma3 or ma2>ma4){
		return false;
	}
	cur=0;
	for(int i=n-1;i>=0;i--){
		while(cur<i){
			if(pre[cur]+pre[i]<ma){
				cur++;
			}
			else{
				break;
			}
		}
		ind[i]=cur;
	}
	cur=0;
	for(int i=n-1;i>=0;i--){
		//cur=0;
		while(cur<i){
			if(pre[cur+1]+pre[i]<=ma3){
				cur++;
			}
			else{
				break;
			}
		}
		ind3[i]=cur;
	}
	cur=0;
	for(int i=0;i<n;i++){
		while(cur<i){
			if(pre[cur]-pre[i]<ma2){
				cur++;
			}
			else{
				break;
			}
		}
		ind2[i]=cur;
	}
	cur=0;
	for(int i=0;i<n;i++){
		while(cur<i){
			if(pre[cur+1]-pre[i]<=ma4){
				cur++;
			}
			else{
				break;
			}
		}
		ind4[i]=cur;
	}

	for(llo i=1;i<n;i++){
		
		if(max(ind[i],ind2[i])<=min(i-1,min(ind3[i],ind4[i]))){
			return true;
		}
	
	}
	

	//ma to ma3 range





	return false;



}
long long find_shortcut(int nn, std::vector<int> l, std::vector<int> d, int c)
{
	n=nn;
	pre[0]=0;
	ss=c;
	for(int i=1;i<n;i++){
		pre[i]=pre[i-1]+l[i-1];
	}


	for(int i=0;i<n;i++){
		it[i]=d[i];
	}
	llo ma=-1;
	llo ma2=-1;
	for(int i=0;i<n;i++){
		if(it[i]>ma){
			ma2=ma;
			ma=it[i];
			continue;
		}
		ma2=max(ma2,it[i]);
	}
	//cout<<ma<<":"<<ma2<<endl;
	ma+=ma2;
	for(int i=0;i<n;i++){
		tt.pb({-pre[i]+it[i],i});	
	}
	sort(tt.begin(),tt.end());
	reverse(tt.begin(),tt.end());


	for(int i=0;i<n;i++){
		tt2.pb({pre[i]+it[i],i});	
	}
	sort(tt2.begin(),tt2.end());

	//reverse(tt2.begin(),tt2.end());
	for(int i=0;i<n;i++){
		ind5[tt[i].b]=i;
	}


	for(int i=0;i<n;i++){
		val[i]=pre[tt[i].b]+it[tt[i].b];
		val2[i]=-pre[tt[i].b]+it[tt[i].b];
		if(i>0){
			val[i]=max(val[i],val[i-1]);
			val2[i]=max(val2[i],val2[i-1]);
		}
	}
	llo low=-1;
	for(int i=52;i>=0;i--){
		if(low+(1LL<<i)<=ma){
			low+=(1LL<<i);
			continue;
		}
		if(check(low+(1LL<<i))==false){
			low+=(1LL<<i);
		}
	}



    return low+1;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...