oj.uz

Submission #294462

# Submission time Handle Problem Language Result Execution time Memory
294462 2020-09-09T00:10:51 Z b00n0rp Towns (IOI15_towns) C++17
100 / 100
 24 ms 800 KB 
#include "towns.h"
#include<bits/stdc++.h>
using namespace std;

#define REP(i,n) for(int i = 0; i < n; i ++)
#define FOR(i,a,b) for(int i = a; i < b; i ++)
#define vi vector<int>
#define pb push_back
#define F first
#define S second

int d0[1005],dA[1005];
int n,A,B,hub,quer;

int dist[1005][1005];

int getD(int i,int j){
	if(dist[i][j] == -1){
		quer++;
		dist[i][j] = getDistance(i,j);
		dist[j][i] = dist[i][j];
	} 
	return dist[i][j];
}

bool samecomp(int i,int j){
	return getD(i,j) != (dA[i]+dA[j]-2*hub);
}

int hubDistance(int N, int sub){
	quer = 0;
	n = N;
	REP(i,n){
		dist[i][i] = 0;
		FOR(j,i+1,n){
			dist[i][j] = -1;
			dist[j][i] = -1;
		}
	}
	A = B = 0;
	d0[0] = 0;
	FOR(i,1,n){
		d0[i] = getD(0,i);
		if(d0[i] > d0[A]) A = i;
	}
	dA[0] = d0[A];
	dA[A] = 0;
	FOR(i,1,n){
		if(i == A) continue;
		dA[i] = getD(A,i);
		if(dA[i] > dA[B]) B = i;
	}
	int R = 1000000000;
	map<int,int> gg; // dist from A from path A-0
	REP(i,n){
		int lol = (dA[0]+dA[i]-d0[i])/2;
		gg[lol]++;
		R = min(R,max(dA[B]-lol,lol));
	}
	int cur = 0;
	hub = -1;
	for(auto x:gg){
		if(cur > n/2) break;
		if(max(dA[B]-x.F,x.F) != R){
			cur += x.S;
			continue;
		}
		int p1 = x.S,p2 = n-cur-x.S;
		cur += p1;
		if(p2 > n/2) continue;
		if(p1 <= n/2) return R;
		hub = x.F;
	}
	if(hub == -1) return -R;
	vi elig;
	REP(i,n){
		if((dA[0]+dA[i]-d0[i])/2 == hub) elig.pb(i);
	}
	vi v1,v2;
	v1.pb(elig[0]);
	FOR(i,1,(int)elig.size()){
		if(samecomp(elig[i],v1.back())){
			v2.pb(elig[i]);
		}
		else{
			v1.pb(elig[i]);
			if(v2.size()){
				v1.pb(v2.back());
				v2.pop_back();
			}
		}
	}
	if(v2.empty()) return R;
	int cnt = (int)v2.size();
	int c = v2[0];
	while(!v1.empty()){
		if(cnt+(int)v1.size() <= n/2) return R;
		if(cnt > n/2) return -R;
		if(samecomp(v1.back(),c)){
			cnt++;
			v1.pop_back();
			if(!v1.empty()) v1.pop_back();
		}
		else v1.pop_back();
	}
	assert(quer <= (7*n)/2-7);
	if(cnt > n/2) return -R;
	return R;
}

Compilation message

towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:30:28: warning: unused parameter 'sub' [-Wunused-parameter]
   30 | int hubDistance(int N, int sub){
      |                        ~~~~^~~

Subtask #1
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 21 ms 768 KB Output is correct
2 Correct 17 ms 768 KB Output is correct
3 Correct 0 ms 384 KB Output is correct
4 Correct 23 ms 768 KB Output is correct
5 Correct 24 ms 768 KB Output is correct

Subtask #2
12.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 21 ms 768 KB Output is correct
2 Correct 21 ms 768 KB Output is correct
3 Correct 23 ms 768 KB Output is correct
4 Correct 23 ms 768 KB Output is correct

Subtask #3
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 768 KB Output is correct
2 Correct 23 ms 800 KB Output is correct
3 Correct 0 ms 384 KB Output is correct
4 Correct 23 ms 768 KB Output is correct

Subtask #4
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 23 ms 768 KB Output is correct

Subtask #5
13.0 / 13.0

# Verdict Execution time Memory Grader output
1 Correct 17 ms 768 KB Output is correct
2 Correct 24 ms 768 KB Output is correct
3 Correct 23 ms 768 KB Output is correct

Subtask #6
39.0 / 39.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 768 KB Output is correct
2 Correct 23 ms 768 KB Output is correct
3 Correct 23 ms 768 KB Output is correct