제출 #289423

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
289423		amoo_safar	Shortcut (IOI16_shortcut)	C++17	38 / 100	2091 ms	93524 KiB

shortcut

#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 = 3010;
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];

	ll C2 = cyc / 2;

	dia2 = Di + Dj + min(c, cyc - c);
	dia = 0;
	for(int k = i + 1; k < j; k++){
		//ll po = lower_bound(ps, ps + n, ps[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])));
	}

	int po = i + 1;
	for(int k = i + 1; k < j; k++){
		while((po < j - 1) && (ps[po + 1] - ps[k] <= C2)) po ++;
		dia = max(dia, d[k] + maxp[k][po] - ps[k]);
		if(po < j - 1){
			dia = max(dia, d[k] + (cyc + maxn[po + 1][j - 1] + ps[k] ));
		}
	}
	po = j - 1;
	for(int k = j - 1; k > i; k--){
		while((po > i + 1) && (ps[k] - ps[po - 1] <= C2)) po --;
		dia = max(dia, d[k] + maxn[po][k] + ps[k]);
		if(po > i + 1){
			dia = max(dia, d[k] + (cyc + maxp[i + 1][po - 1] - 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;
}

컴파일 시 표준 에러 (stderr) 메시지

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

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