Submission #430492

#	Time	Username	Problem	Language	Result	Execution time	Memory
430492		Blagojce	Rail (IOI14_rail)	C++11	30 / 100	109 ms	512 KiB

rail

#include <bits/stdc++.h>
#define fr(i, n, m) for(int i = (n); i < (m); i ++)
#define pb push_back
#define st first
#define nd second
#define pq priority_queue
#define all(x) begin(x), end(x)
  
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;
typedef pair<ll, ll> pll;
 
const ll inf = 1e18;
const int i_inf = 1e9;
const ll mod = 1e9+7;
 
mt19937 _rand(time(NULL));
const int mxn = 5005;

#include "rail.h"

int d0[mxn];
int d1[mxn];
/*
int getDistance(int i, int j){
	cout<<"? "<<i<<' '<<j<<endl;
	int ret;
	cin >> ret;
	
	return ret;
}
*/


vector<int> L;
vector<int> R;
void findLocation(int N, int first, int location[], int stype[]){
	location[0] = first;
	stype[0] = 1;
	
	
	if(N == 1){
		return;
	}
	
	fr(i, 1, N){
		d0[i] =  getDistance(0, i);
	}
	int minD = 1e9;
	int p = 0;
	fr(i, 1, N){
		if(d0[i] < minD){
			minD = d0[i];
			p = i;
		}
	}
	
	d1[0] = d0[p];
	
	fr(i, 1, N){
		if(i == p) continue;
		d1[i] = getDistance(p, i);
	}
	
	
	
	
	fr(i, 0, N){
		if(i == p || i == 0) continue;
		
		if(d0[i] == d1[i] + d0[p]){
			L.pb(i);
		}
		else{
			R.pb(i);
		}
	}
	
	
	
	
	
	sort(all(R), [](const int &i, const int &j){
		return d0[i] < d0[j];
	});
	location[p] = location[0] + d0[p];
	
	stype[p] = 2;
	
	if(N == 2){
		return;
	}
	
	
	queue<int> P;
	P.push(p);
	
	
	for(auto u : R){
		while(!P.empty()){
			int piv = P.front();
			int w = getDistance(piv, u);
			if(d0[u] == d0[piv] + w){
				location[u] = location[piv] - w;
				stype[u] = 1;
				break;
			}
			else{
				P.pop();
			}
		}
		if(P.empty()){
			P.push(u);
			location[u] = location[0] + d0[u];
			stype[u] = 2;
		}
	}
	
	
	int p2;
	minD = 1e9;
	fr(i, 0, N){
		if(i == p) continue;
		if(d1[i] < minD){
			minD = d1[i];
			p2 = i;
		}
		
	}
	L.pb(0);
	
	sort(all(L), [](const int &i, const int &j){
		return d1[i] < d1[j];
	});
	
	
	location[p2] = location[p] - d1[p2];
	stype[p2] = 1;
	
	
	queue<int> P2;
	P2.push(p2);
	
	for(auto u : L){
		if(u == p2) continue;
		
		while(!P2.empty()){
			int piv = P2.front();
			int w = getDistance(piv, u);
			if(d1[u] == d1[piv] + w){
				location[u] = location[piv] + w;
				stype[u] = 2;
				break;
			}
			else{
				P2.pop();
			}
		}
		if(P2.empty()){
			location[u] = location[p] - d1[u];
			stype[u] = 1;
			P2.push(u);
		}
		
	}
}
/*
int main(){
	int loc[mxn];
	int sty[mxn];
	findLocation(4, 3, loc, sty);
	fr(i, 0, 5){
		cout<<loc[i]<<' ';
	}
	cout<<endl;
	
	fr(i, 0, 5){
		cout<<sty[i]<<' ';
	}
	cout<<endl;
	
}
*/

/*
0 1 2 3 4 5 6 7
(   ) (   ) 
3   0 1   2
*/
/*3
5
8
9
2
5
6
5
8
1
*/

• Subtask 1: 8 / 8
• Subtask 2: 22 / 22
• Subtask 3: 0 / 26
• Subtask 4: 0 / 44