Submission #795441

#	Time	Username	Problem	Language	Result	Execution time	Memory
795441		Josia	Towns (IOI15_towns)	C++17	35 / 100	15 ms	360 KiB

towns

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


int hubDistance(int N, int sub) {
	int d1=0, v1 = 0;
	for (int i = 1; i<N; i++) {
		int d = getDistance(0, i);
		if (d>d1) {
			d1 = d;
			v1 = i;
		}
	}

	vector<int> dists1(N);

	int d2=0, v2=v1;
	for (int i = 0; i<N; i++) {
		int d = getDistance(v1, i);
		dists1[i] = d;
		if (d>d2) {
			d2 = d;
			v2 = i;
		}
	}

	vector<int> dists2(N);
	for (int i = 0; i<N; i++) {
		int d = getDistance(v2, i);
		dists2[i] = d;
	}


	map<int, vector<int>> joinedDiam;


	int R = 1e9;
	for (int i = 0; i<N; i++) {
		int rem = (dists1[i]+dists2[i]-d2)/2;

		joinedDiam[dists1[i]-rem].push_back(i);

		R = min(R, max(dists1[i]-rem, dists2[i]-rem));
	}

	vector<int> possHub;

	for (int i = 0; i<N; i++) {
		int rem = (dists1[i]+dists2[i]-d2)/2;
		if (max(dists1[i]-rem, dists2[i]-rem) == R) possHub.push_back(dists1[i]-rem);
	}

	int mult = -1;
	for (int hub: possHub) {
		int h1=0, h2=0;
		bool poss = 1;
		for (auto i: joinedDiam) {
			if (i.first < hub) h1+=i.second.size();
			if (i.first > hub) h2+=i.second.size();
			if (i.first == hub) if ((int)i.second.size() > N/2) poss = 0;
		}
		if (h1 > N/2 || h2 > N/2) poss = 0;

		if (poss) mult = 1;
	}


	return R*mult;
}

Compilation message (stderr)

towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:59:41: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   59 |    if (i.first < hub) h1+=i.second.size();
      |                                         ^
towns.cpp:60:41: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   60 |    if (i.first > hub) h2+=i.second.size();
      |                                         ^
towns.cpp:6:28: warning: unused parameter 'sub' [-Wunused-parameter]
    6 | int hubDistance(int N, int sub) {
      |                        ~~~~^~~

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