제출 #400850

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
400850		kshitij_sodani	Shortcut (IOI16_shortcut)	C++14	31 / 100	2077 ms	38664 KiB

shortcut

#pragma GCC optimize("Ofast,unroll-loops")
#include <bits/stdc++.h>
using namespace std;
typedef long long llo;
#define mp make_pair
#define pb push_back
#define a first 
#define b second
//#define endl '\n'
 
#include "shortcut.h"
//vector<pair<llo,llo>> adj[1000001];
llo dist[6001][6001];
vector<pair<llo,llo>> adj[1000001];
llo dp[6001][6001];

void dfs(llo no,llo par=-1,llo no2=-1,llo lev=0){
	dist[no2][no]=lev;
	for(auto j:adj[no]){
		if(j.a!=par){
			dfs(j.a,no,no2,lev+j.b);
		}
	}
}
llo cc;

long long find_shortcut(int n, std::vector<int> aa, std::vector<int> bb, int ccc)
{
	cc=ccc;
	for(int i=0;i<n-1;i++){
		adj[i].pb({i+1,aa[i]});
		adj[i+1].pb({i,aa[i]});
	}
	for(int i=0;i<n;i++){
		adj[i].pb({n+i,bb[i]});
		adj[n+i].pb({i,bb[i]});
	}
	for(int i=0;i<2*n;i++){
		dfs(i,-1,i);
	}
	/*for(int i=0;i<2*n;i++){
		for(int j=0;j<2*n;j++){
			cout<<dist[i][j]<<",";
		}
		cout<<endl;
	}*/

	for(int i=n-1;i>=0;i--){
		//llo xx=0;
		for(int j=i+1;j<n;j++){
			//xx+=aa[j-1];
			dp[i][j]=dist[i+n][j+n];
			dp[i][j]=max(dp[i][j],dp[i][j-1]);
			dp[i][j]=max(dp[i][j],dp[i+1][j]);
		}
	}

	llo ans=1e18;
	for(int i=0;i<n;i++){
		for(int j=i+1;j<n;j++){
			llo ma=0;
			llo ind=j+1;
		//	llo su=0;
			//llo su2=0;
			//llo su3=dist[i][j];

			vector<pair<int,int>> ss;
			llo yy=0;
			for(int k=j-1;k>=i;k--){
				/*if(k>0){
					su+=aa[k];
				}*/
 				
				while(ind-1>k){
					if(dist[k][i]+dist[ind-1][j]+cc<=dist[k][ind-1]){
						yy=max(yy,dist[k][i]+dist[ind-1][j]+cc);
						ind--;
					}
					else{
						break;
					}
				}
				//ss.pb({i,ind});
 				/*if(ind<=j){
 					ma=max(ma,yy+dist[i][k+n]);
 				}*/
				//ma=max(ma,eval(i,ind));
				if(ind>k){
					ma=max(ma,dp[k][ind-1]);
					//ss.pb({i,ind-1});
					//ma=max(ma,eval(ind,i));
				}
			}
			//if(cc<=dist[i][j]){
				llo ma3=0;
				for(int k=i;k>=0;k--){
					ma3=max(ma3,dist[k+n][i]);
				}
				llo ma4=0;
				for(int k=j;k<n;k++){
					ma4=max(ma4,dist[k+n][j]);
				}
				ma=max(ma,ma3+ma4+min((llo)cc,dist[i][j]));
			//}
			llo ma2=0;
			for(int k=i;k<=j;k++){
				llo xx=min(dist[i][k]+cc,dist[k][j]);
				ma2=max(ma2,xx+bb[k]);
			}
			ma=max(ma,ma2);
			for(int k=j+1;k<n;k++){
				ma=max(ma,dist[k][j]+bb[k]+ma2);
			}
			ma2=0;
			for(int k=j;k>=i;k--){
				llo xx=min(dist[k][j]+cc,dist[k][i]);
				ma2=max(ma2,xx+bb[k]);
			}
			ma=max(ma,ma2);
			for(int k=i-1;k>=0;k--){
				ma=max(ma,bb[k]+ma2+dist[k][i]);
			}
			//if(cc<=su3){
			/*for(int k=0;k<=i;k++){
				for(int l=j;l<n;l++){
					k+=n;
					l+=n;
					ma=max(ma,min(dist[k][l],min(dist[k][i]+cc+dist[j][l],dist[k][j]+cc+dist[i][l])));
					k-=n;
					l-=n;				
				}
			}*/
		//}
		/*	for(int k=0;k<=i;k++){
				for(int l=j;l<n;l++){
					k+=n;
					l+=n;
					ma=max(ma,min(dist[k][l],min(dist[k][i]+cc+dist[j][l],dist[k][j]+cc+dist[i][l])));
					k-=n;
					l-=n;				
				}
			}*/
			/*for(int k=i;k<=j;k++){
				for(int l=j;l<n;l++){
					k+=n;
					l+=n;
					ma=max(ma,min(dist[k][l],min(dist[k][i]+cc+dist[j][l],dist[k][j]+cc+dist[i][l])));
					k-=n;
					l-=n;
				}
			}
			for(int k=i;k<=j;k++){
				for(int l=0;l<=i;l++){
					k+=n;
					l+=n;
					ma=max(ma,min(dist[k][l],min(dist[k][i]+cc+dist[j][l],dist[k][j]+cc+dist[i][l])));
					k-=n;
					l-=n;
				}
			}*/
			for(auto kk:ss){
				int k=kk.a+n;
				int l=kk.b+n;
				ma=max(ma,min(dist[k][l],min(dist[k][i]+cc+dist[j][l],dist[k][j]+cc+dist[i][l])));
			}
			for(int k=i;k<=j;k++){
				for(int l=k;l<=j;l++){
					k+=n;
					l+=n;
					ma=max(ma,min(dist[k][l],min(dist[k][i]+cc+dist[j][l],dist[k][j]+cc+dist[i][l])));
					k-=n;
					l-=n;
				}
			}
		
 			ma=max(ma,dp[0][i]);
			ma=max(ma,dp[j][n-1]);
 
			/*for(int k=n;k<2*n;k++){
				for(int l=k;l<2*n;l++){
					ma=max(ma,min(dist[k][l],min(dist[k][i]+cc+dist[j][l],dist[k][j]+cc+dist[i][l])));		
					
				}
			}*/
			/*if(ma==0){
				cout<<i<<":"<<j<<endl;
			}*/
			ans=min(ans,ma);
		}
	}
 
 
 
 
    return ans;
}

• Subtask 1: 9 / 9
• Subtask 2: 14 / 14
• Subtask 3: 8 / 8
• Subtask 4: 0 / 7
• Subtask 5: 0 / 33
• Subtask 6: 0 / 22
• Subtask 7: 0 / 4
• Subtask 8: 0 / 3