제출 #400874

#제출 시각아이디문제언어결과실행 시간메모리
400874kshitij_sodaniShortcut (IOI16_shortcut)C++14
38 / 100
2117 ms353768 KiB
#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;
int n;
std::vector<int> aa;
std::vector<int> bb;
llo eval(llo i,llo j){
	llo ma=0;
			llo ind=j+1;
			vector<pair<int,int>> ss;
			llo yy=0;
			for(int k=j-1;k>=i;k--){

				while(ind-1>k){
					if(dist[k][i]+dist[ind-1][j]+cc<=dist[k][ind-1]){
						yy=max(yy,dist[ind-1+n][j]+cc);
						ind--;
					}
					else{
						break;
					}
				}
 				if(ind<=j){
 					ma=max(ma,yy+bb[k]+dist[k][i]);
 				}
				if(ind>k){
					ma=max(ma,dp[k][ind-1]);
				}
			}
				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]);
			}

			/*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]);
		return ma;
			//ans=min(ans,ma);
}
long long find_shortcut(int nn, std::vector<int> aaa, std::vector<int> bbb, int ccc)
{
		n=nn;
	aa=aaa;
	bb=bbb;
	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++){
			ans=min(ans,eval(i,j));
		}
	}
 
 
 
 
    return ans;
}
#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...