제출 #1018806

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1018806		AmirAli_H1	도시들 (IOI15_towns)	C++17	48 / 100	11 ms	1120 KiB

towns

// In the name of Allah

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

typedef		long long int			ll;
typedef		long double				ld;
typedef		pair<int, int>			pii;
typedef		pair<ll, ll>			pll;
typedef		complex<ld>				cld;

#define		all(x)					(x).begin(),(x).end()
#define		len(x)					((ll) (x).size())
#define		F						first
#define		S						second
#define		pb						push_back
#define		sep						' '
#define		endl					'\n'
#define		Mp						make_pair
#define		kill(x)					cout << x << '\n', exit(0)
#define		set_dec(x)				cout << fixed << setprecision(x);
#define		file_io(x,y)			freopen(x, "r", stdin); freopen(y, "w", stdout);
mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

int n;
const int maxn = 110 + 4;
ll dis[maxn][maxn];
ll D1[maxn], D2[maxn];
ll ans, R, dx; bool f1, f2;
int M[maxn], p[maxn], sz[maxn];

int get(int a) {
	return (p[a] == a) ? a : p[a] = get(p[a]);
}

void merge(int a, int b) {
	a = get(a); b = get(b);
	if (a == b) return ;
	if (sz[a] > sz[b]) swap(a, b);
	p[a] = b; sz[b] += sz[a]; sz[a] = 0;
}

int get_dis(int u, int v) {
	if (dis[u][v] == -1) {
		dis[u][v] = dis[v][u] = getDistance(u, v);
	}
	return dis[u][v];
}

void solve3() {
	if (f1) {
		int t2 = 0, t3 = 0; fill(M, M + n, 0);
		iota(p, p + n, 0); fill(sz, sz + n, 1);
		
		for (int i = 0; i < n; i++) {
			ll x = (D1[i] + D2[i] - dx) / 2;
			ll d = D1[i] - x;
			if (d == R) M[i] = 1;
			else if (d < R) t2++;
			else t3++;
		}
		
		for (int i = 0; i < n; i++) {
			for (int j = i + 1; j < n; j++) {
				ll x1 = (D1[i] + D2[i] - dx) / 2;
				ll x2 = (D1[j] + D2[j] - dx) / 2;
				ll d = (x1 + x2 - get_dis(i, j)) / 2;
				if (d > 0) merge(i, j);
			}
		}
		
		int mx = *max_element(sz, sz + n);
		if (2 * max({mx, t2, t3}) <= n) return ;
	}
	if (f2) {
		int t2 = 0, t3 = 0; fill(M, M + n, 0);
		iota(p, p + n, 0); fill(sz, sz + n, 1);
		
		for (int i = 0; i < n; i++) {
			ll x = (D1[i] + D2[i] - dx) / 2;
			ll d = D2[i] - x;
			if (d == R) M[i] = 1;
			else if (d < R) t2++;
			else t3++;
		}
		
		for (int i = 0; i < n; i++) {
			for (int j = i + 1; j < n; j++) {
				ll x1 = (D1[i] + D2[i] - dx) / 2;
				ll x2 = (D1[j] + D2[j] - dx) / 2;
				ll d = (x1 + x2 - get_dis(i, j)) / 2;
				if (d > 0) merge(i, j);
			}
		}
		
		int mx = *max_element(sz, sz + n);
		if (2 * max({mx, t2, t3}) <= n) return ;
	}
	
	ans = -ans;
	return ;
}

void solve4() {
	if (f1) {
		int t1 = 0, t2 = 0, t3 = 0;
		for (int i = 0; i < n; i++) {
			ll x = (D1[i] + D2[i] - dx) / 2;
			ll d = D1[i] - x;
			if (d == R) t1++;
			else if (d < R) t2++;
			else t3++;
		}
		if (2 * max({t1, t2, t3}) <= n) return ;
	}
	if (f2) {
		int t1 = 0, t2 = 0, t3 = 0;
		for (int i = 0; i < n; i++) {
			ll x = (D1[i] + D2[i] - dx) / 2;
			ll d = D2[i] - x;
			if (d == R) t1++;
			else if (d < R) t2++;
			else t3++;
		}
		if (2 * max({t1, t2, t3}) <= n) return ;
	}
	
	ans = -ans;
	return ;
}

int hubDistance(int N, int sub) {
	n = N;
	
	for (int i = 0; i < n; i++) {
		for (int j = 0; j < n; j++) {
			if (i == j) dis[i][j] = 0;
			else dis[i][j] = -1;
		}
	}
	
	ll mx = 0; int u1 = 0;
	for (int i = 0; i < n; i++) {
		if (i == 0) continue;
		ll d = get_dis(0, i);
		if (d >= mx) {
			mx = d; u1 = i;
		}
	}
	
	mx = 0; int u2 = u1;
	for (int i = 0; i < n; i++) {
		if (i == u1) {
			D1[i] = 0;
			continue;
		}
		D1[i] = get_dis(u1, i);
		if (D1[i] >= mx) {
			mx = D1[i]; u2 = i;
		}
	}
	
	for (int i = 0; i < n; i++) {
		if (i == u2) {
			D2[i] = 0;
			continue;
		}
		D2[i] = get_dis(u2, i);
	}
	
	dx = D1[u2]; R = dx;
	for (int i = 0; i < n; i++) {
		ll x = (D1[i] + D2[i] - dx) / 2;
		R = min(R, max(D1[i] - x, D2[i] - x));
	}
	ans = R;
	
	f1 = 0, f2 = 0;
	for (int i = 0; i < n; i++) {
		ll x = (D1[i] + D2[i] - dx) / 2;
		if (D1[i] - x == R) f1 = 1;
		if (D2[i] - x == R) f2 = 1;
	}
	
	if (sub == 3) solve3();
	else if (sub == 4) solve4();
	return ans;
}

컴파일 시 표준 에러 (stderr) 메시지

towns.cpp: In function 'int get_dis(int, int)':
towns.cpp:48:17: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
   48 |  return dis[u][v];
      |         ~~~~~~~~^
towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:188:9: warning: conversion from 'll' {aka 'long long int'} to 'int' may change value [-Wconversion]
  188 |  return ans;
      |         ^~~

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