Submission #289413

#TimeUsernameProblemLanguageResultExecution timeMemory
289413amoo_safarShortcut (IOI16_shortcut)C++17
38 / 100
2091 ms93176 KiB
#include "shortcut.h"

#include <bits/stdc++.h>
#define F first
#define S second

using namespace std;

typedef long long ll;

typedef pair<ll, ll> pll;

const int N = 3000;
const ll Inf = 1e18;

ll ps[N];
ll maxp[N][N], maxn[N][N];
ll pre[N], suf[N];
ll d[N];
ll n, c;


pll Solve(ll i, ll j){

	ll Di, Dj, dia, dia2;
	ll cyc;

	Di = ps[i] + maxn[0][i];
	Dj = maxp[j][n - 1] - ps[j];
	cyc = c + ps[j] - ps[i];

	dia2 = Di + Dj + min(c, cyc - c);
	dia = 0;
	for(int k = i + 1; k < j; k++){
		for(int k2 = k + 1; k2 < j; k2++){
			dia = max(dia, d[k] + d[k2] + min(ps[k2] - ps[k], cyc - (ps[k2] - ps[k]) ) );
		}
		dia = max(dia, Di + d[k] + min(ps[k] - ps[i], cyc - (ps[k] - ps[i])));
		dia2 = max(dia2, Dj + d[k] + min(ps[j] - ps[k], cyc - (ps[j] - ps[k])));
	}
	dia2 = max(dia2, suf[j]);
	dia = max(dia, pre[i]);
	return make_pair(dia, dia2);
}

long long find_shortcut(int _n, vector<int> _l, vector<int> _d, int _c){
	n = _n; c = _c;
	for(int i = 0; i < n; i++) d[i] = _d[i];
	ps[0] = 0;
	for(int i = 1; i < n; i++) ps[i] = ps[i - 1] + _l[i - 1];

	// Calc Maxp
	for(int i = 0; i < n; i++) maxp[i][i] = d[i] + ps[i];
	for(int i = 0; i < n; i++) maxn[i][i] = d[i] - ps[i];
	for(int ln = 2; ln <= n; ln++){
		for(int i = 0; i + ln <= n; i++){
			int j = i + ln - 1;
			maxp[i][j] = max(maxp[i + 1][j], maxp[i][j - 1]);
			maxn[i][j] = max(maxn[i + 1][j], maxn[i][j - 1]);
		}
	}
	// Calc pre
	pre[0] = d[0];
	for(int i = 1; i < n; i++){
		pre[i] = max(pre[i - 1], d[i] + ps[i] + maxn[0][i - 1]);
	}	
	suf[n - 1] = d[n - 1];
	for(int i = n - 2; i >= 0; i--){
		suf[i] = max(suf[i + 1], d[i] - ps[i] + maxp[i + 1][n - 1]);
	}



	ll dia, dia2, ans = Inf;
	pll res;
	vector<pll> V;
	for(int i = 0; i + 1 < n; i++){
		int L = i, R = n, mid;
		while(L + 1 < R){
			mid = (L + R) >> 1;
			res = Solve(i, mid);
			if(res.F <= res.S) L = mid;
			else R = mid;
		}
		pll best = {Inf, Inf};
		dia = Inf;
		if(L != i){
			res = Solve(i, L);
			if(max(best.F, best.S) > max(res.F, res.S))
				best = res;
		}

		if(R != n){
			res = Solve(i, R);
			if(max(best.F, best.S) > max(res.F, res.S))
				best = res;
		}
		assert(best != pll(Inf, Inf));
		ans = min(ans, max(best.F, best.S));
		V.push_back(best);
		/*
		V.clear();
		for(int j = i + 1; j < n; j++){
			res = Solve(i, j);
			dia = max({res.first, res.second});
			ans = min(ans, dia);
			V.push_back(res);
		}
		int sz = (int) V.size();
		for(int j = 0; j + 1 < sz; j++){
			assert(V[j].first <= V[j + 1].first);
			assert(V[j].second >= V[j + 1].second);	
		}
		*/
	}
	/*
	int sz = (int) V.size();
	for(int j = 0; j + 1 < sz; j++){
		assert(V[j].first <= V[j + 1].first);
		assert(V[j].second >= V[j + 1].second);	
	}
	*/
	return ans;
}

Compilation message (stderr)

shortcut.cpp: In function 'long long int find_shortcut(int, std::vector<int>, std::vector<int>, int)':
shortcut.cpp:74:5: warning: variable 'dia' set but not used [-Wunused-but-set-variable]
   74 |  ll dia, dia2, ans = Inf;
      |     ^~~
shortcut.cpp:74:10: warning: unused variable 'dia2' [-Wunused-variable]
   74 |  ll dia, dia2, ans = Inf;
      |          ^~~~
#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...