#include "towns.h"
#include <bits/stdc++.h>
using namespace std;
#define forn(i,n) for(int i=0;i<n;++i)
#define pb push_back
#define all(x) x.begin(), x.end()
#define pi pair<int,int>
#define f first
#define s second
const int inf=1e9+9;
const int _N=555;
int d[_N][_N];
struct DSU {
vector<int> p,sz;
DSU(int n) {forn(i,n)p.pb(i),sz.pb(1);}
int get(int u) {return p[u]==u?u:get(p[u]);}
void uni(int u, int v){
u=get(u),v=get(v); if (u==v)return; if (sz[u]<sz[v]) swap(u,v);
p[v]=u;sz[u]+=sz[v];
}
};
int p3(int n) {
forn(i,n) {
forn(j,i) {
int x=getDistance(i,j);
d[i][j]=d[j][i]=x;
}
}
int f=0,s=0;
forn(i,n) if (d[0][i]>d[0][f]) f=i;
forn(i,n) if (d[f][i]>d[f][s]) s=i;
int ans=inf;
map<int,int> cnt;
map<int,vector<int>> q;
vector<int> ok(n);
set<int> s1,s2;
forn(i,n) {
if (i==f || i==s) continue;
int x=d[f][i], y=d[s][i], z=d[f][s];
int A=(x-y+z)/2;
int B=z-A;
int C=x-A;
ans=min(ans,max(A,B));
ok[i]=C;
cnt[A]++;
q[A].pb(i);
s1.insert(A), s2.insert(B);
}
int left=1;
vector<int> x,y;
for(auto&t:s1)x.pb(t);
for(auto&t:s2)y.pb(t);
reverse(all(y));
forn(i,s1.size()) {
if (max(x[i],y[i])==ans) {
int right=n-left-cnt[x[i]];
int mx=max(left,right);
DSU dsu(n);
for(auto&u:q[x[i]]) {
for(auto&v:q[x[i]]) {
int dist = d[u][v];
if (dist < ok[u]+ok[v]) dsu.uni(u,v);
}
}
forn(j,n) mx=max(mx,dsu.sz[j]);
if (mx<=n/2) return ans;
}
left+=cnt[x[i]];
}
return -ans;
}
int p5(int n) {
vector<int> a(n), b(n), c(n);
int f=0;
forn(i,n) {
a[i]=getDistance(0,i);
if (a[i]>a[f]) f=i;
}
int s=0;
forn(i,n) {
b[i]=getDistance(f,i);
if (b[i]>b[s]) s=i;
}
forn(i,n) {
c[i]=getDistance(s,i);
}
set<int> st;
vector<int> d(n);
map<int,vector<int>> ok;
forn(i,n) {
if (i==f || i==s) continue;
int x=b[i], y=c[i], z=b[s];
int A=(x-y+z)/2;
d[i]=x-A;
st.insert(A);
ok[A].pb(i);
}
int ans=inf;
for(auto&x:st) {
int y=b[s]-x;
ans=min(ans,max(x,y));
}
int l=1;
for(auto&x:st) {
int y=b[s]-x;
if (max(x,y)!=ans) {
l+=ok[x].size();
continue;
}
int r=n-ok[x].size()-l;
if (max(l,r)>n/2) continue;
int t=ok[x].size();
if (t<=n/2) return ans;
vector<pi> a,b;
for(auto&v:ok[x]) a.pb({v,1});
while (a.size()>1) {
int k = a.size();
for (int i=0; i+1<a.size(); i+=2) {
int z = getDistance(a[i].f,a[i+1].f);
int x = d[a[i].f], y = d[a[i+1].f];
if (z==x+y) {
if (a[i].s > a[i+1].s) b.pb(a[i]);
else b.pb(a[i+1]);
continue;
}
b.pb({a[i].f,a[i].s+a[i+1].s});
}
if (k&1) b.pb(a.back());
a=b;
b.clear();
}
int u=a[0].f;
int sz=0;
for(auto&v:ok[x]) {
int z = getDistance(u,v);
int x = c[u], y = c[v];
if (z<x+y) ++sz;
}
if (sz<=n/2) return ans;
}
return -ans;
}
int hubDistance(int n, int sub) {
if (sub<=2 || sub>=4) return p5(n);
if (sub==3) return p3(n);
return 0;
}
Compilation message
towns.cpp: In function 'int p3(int)':
towns.cpp:4:32: warning: comparison of integer expressions of different signedness: 'int' and 'std::set<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
4 | #define forn(i,n) for(int i=0;i<n;++i)
......
61 | forn(i,s1.size()) {
| ~~~~~~~~~~~
towns.cpp:61:3: note: in expansion of macro 'forn'
61 | forn(i,s1.size()) {
| ^~~~
towns.cpp: In function 'int p5(int)':
towns.cpp:99:15: warning: declaration of 'd' shadows a global declaration [-Wshadow]
99 | vector<int> d(n);
| ^
towns.cpp:13:5: note: shadowed declaration is here
13 | int d[_N][_N];
| ^
towns.cpp:120:21: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
120 | l+=ok[x].size();
| ^
towns.cpp:123:25: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
123 | int r=n-ok[x].size()-l;
| ~~~~~~~~~~~~~~^~
towns.cpp:125:21: warning: conversion from 'std::vector<int>::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
125 | int t=ok[x].size();
| ~~~~~~~~~~^~
towns.cpp:127:16: warning: declaration of 'a' shadows a previous local [-Wshadow]
127 | vector<pi> a,b;
| ^
towns.cpp:82:15: note: shadowed declaration is here
82 | vector<int> a(n), b(n), c(n);
| ^
towns.cpp:127:18: warning: declaration of 'b' shadows a previous local [-Wshadow]
127 | vector<pi> a,b;
| ^
towns.cpp:82:21: note: shadowed declaration is here
82 | vector<int> a(n), b(n), c(n);
| ^
towns.cpp:130:21: warning: conversion from 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
130 | int k = a.size();
| ~~~~~~^~
towns.cpp:131:24: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
131 | for (int i=0; i+1<a.size(); i+=2) {
| ~~~^~~~~~~~~
towns.cpp:133:13: warning: declaration of 'x' shadows a previous local [-Wshadow]
133 | int x = d[a[i].f], y = d[a[i+1].f];
| ^
towns.cpp:117:12: note: shadowed declaration is here
117 | for(auto&x:st) {
| ^
towns.cpp:133:28: warning: declaration of 'y' shadows a previous local [-Wshadow]
133 | int x = d[a[i].f], y = d[a[i+1].f];
| ^
towns.cpp:118:9: note: shadowed declaration is here
118 | int y=b[s]-x;
| ^
towns.cpp:149:11: warning: declaration of 'x' shadows a previous local [-Wshadow]
149 | int x = c[u], y = c[v];
| ^
towns.cpp:117:12: note: shadowed declaration is here
117 | for(auto&x:st) {
| ^
towns.cpp:149:21: warning: declaration of 'y' shadows a previous local [-Wshadow]
149 | int x = c[u], y = c[v];
| ^
towns.cpp:118:9: note: shadowed declaration is here
118 | int y=b[s]-x;
| ^
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
10 ms |
340 KB |
Output is correct |
2 |
Correct |
8 ms |
368 KB |
Output is correct |
3 |
Correct |
0 ms |
212 KB |
Output is correct |
4 |
Correct |
11 ms |
356 KB |
Output is correct |
5 |
Correct |
14 ms |
356 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
468 KB |
Output is correct |
2 |
Correct |
15 ms |
576 KB |
Output is correct |
3 |
Correct |
0 ms |
340 KB |
Output is correct |
4 |
Correct |
12 ms |
576 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
8 ms |
340 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
212 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |