Submission #399727

#	Time	Username	Problem	Language	Result	Execution time	Memory
399727		peuch	Towns (IOI15_towns)	C++17	35 / 100	22 ms	420 KiB

towns

#include "towns.h"
#include<bits/stdc++.h>
using namespace std;

const int MAXN = 1e6;

int dist[3][MAXN];

int hubDistance(int N, int sub) {
	int p1 = 0, p2 = 0;
	int maxi = 0;
	for(int i = 0; i < N; i++){
		int aux = getDistance(0, i);
		if(aux > maxi) p1 = i, maxi = aux;
	}
	maxi = 0;
	for(int i = 0; i < N; i++){
		dist[0][i] = getDistance(p1, i);
		if(dist[0][i] > maxi) maxi = dist[0][i], p2 = i;
	}
	for(int i = 0; i < N; i++)
		dist[1][i] = getDistance(p2, i);
	
	int diam = dist[0][p2];
	int ans = MAXN;
	for(int i = 0; i < N; i++){
		int x = (diam + dist[0][i] - dist[1][i]) / 2;
		ans = min(ans, max(x, diam - x));
	}
	
	if(sub <= 2) return ans;
	
	int lSize = 0, rSize = 0;
	vector<int> lMid, rMid;
	for(int i = 0; i < N; i++){
		int x = (diam + dist[0][i] - dist[1][i]) / 2;
		if(x < min(ans, diam - ans)) lSize++;
		else if(x == min(ans, diam - ans)) lMid.push_back(i); 
		else if(x == max(ans, diam - ans)) rMid.push_back(i);
		else rSize++;
	}
	
	if(lSize > N / 2 || rSize > N / 2) return -ans;
	
	if(lMid.size() > N / 2){
		return -ans;
	}
	else if(rMid.size() > N / 2){
		return -ans;
	}
	return ans;
}

Compilation message (stderr)

towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:45:17: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   45 |  if(lMid.size() > N / 2){
      |     ~~~~~~~~~~~~^~~~~~~
towns.cpp:48:22: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   48 |  else if(rMid.size() > N / 2){
      |          ~~~~~~~~~~~~^~~~~~~

• Subtask 1: 13 / 13
• Subtask 2: 12 / 12
• Subtask 3: 0 / 13
• Subtask 4: 10 / 10
• Subtask 5: 0 / 13
• Subtask 6: 0 / 39