제출 #1224571

#제출 시각아이디문제언어결과실행 시간메모리
1224571ByeWorldFinding Routers (IOI20_routers)C++20
98.92 / 100
1091 ms365108 KiB
#include "routers.h"
#include <bits/stdc++.h>
#define us use_detector
#define ll long long
#define se second
#define fi first
#define pb push_back
#define lf (id<<1)
#define rg ((id<<1)|1)
#define md ((l+r)>>1)
using namespace std;

void chmn(auto &a, auto b){ a = min(a, b); }
void chmx(auto &a, auto b){ a = max(a, b); }
const int MAXN = 1010;

int n, ma;
int a[1010], dis[1010]; // i ke i-1
set<int> le[MAXN], ri[MAXN];

void find(int idx){
	int mx = 0, mn = ma+1;
	for(int i=0; i<=idx-1; i++) chmx(mx, a[i]);
	for(int i=idx+1; i<=n-1; i++) 
		if(a[i] != -1) chmn(mn, a[i]);

	int l=mx+1, r=mn-1, mid=0;
	if(!le[idx].empty()) chmx(l, *--le[idx].end());
	if(!ri[idx].empty()) chmn(r, *ri[idx].begin());
	int cnt=l; l++;

	while(l<=r){
		mid = md;
		int ret = us(mid); 
		for(int i=1; i<=ret-1; i++) ri[i].insert(mid);
		for(int i=ret+1; i<=n-1; i++) le[i].insert(mid);

		if(ret < idx) cnt = mid, l = mid+1;
		else r = mid-1;
	}
	dis[idx] = 2*cnt;
}

void solve(int le, int ri){
	if(le>ri) return;
	int mid = (le+ri)>>1;
	find(mid);
	solve(le, mid-1); solve(mid+1, ri);
}
std::vector<int> find_routers(int MA, int N, int q) {
	n = N; ma = MA;
	for(int i=1; i<=n-1; i++) a[i] = -1;

	find(n-1);
	solve(1, n-2);
	// if(n/2 != 0){
	// 	find(n/2);
	// 	for(int i=n/2+1; i<=n-1; i++) find(i);
	// 	for(int i=n/2-1; i>=1; i--) find(i);
	// } else {
	// 	for(int i=1; i<=n-1; i++) find(i);
	// }
	vector<int> ans; ans.pb(0);
	for(int i=1; i<=n-1; i++){
		a[i] = dis[i]-a[i-1];
		// cout << a[i] << " a\n";
		ans.pb(a[i]);
		assert(a[i] != -1);
	}
	// for(auto in : ans) cout << in << " in\n";
	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...