#include "towns.h"
#include <bits/stdc++.h>
using namespace std;
using ar2 = array<int,2>;
const int mxN = 115;
int n, vis[mxN];
int mem[mxN][mxN];
map<int,int> M;
int dis(int a, int b){
if(a==b) return 0;
if(mem[a][b]!=-1) return mem[a][b];
if(mem[b][a]!=-1) return mem[b][a];
return mem[a][b]=mem[b][a]=getDistance(a,b);
}
int p[mxN], sz[mxN];
int findSet(int i) { return p[i]==i?i:p[i]=findSet(p[i]); }
bool isSameSet(int i, int j) { return findSet(i)==findSet(j); }
void unionSet(int i, int j){
int x = findSet(i), y = findSet(j);
if(x==y) return;
if(sz[x]<sz[y]) swap(x,y);
p[y]=x; sz[x]+=sz[y];
}
int hubDistance(int N, int sub) {
n = N; ar2 mx = {0,0};
int a = 0, b = 0; M.clear();
memset(mem,-1,sizeof(mem));
for(int i = 0; i < n; i++)
mx=max(mx,{dis(a,i),i});
a = mx[1]; mx = {0,0};
for(int i = 0; i < n; i++)
mx=max(mx,{dis(a,i),i});
b = mx[1];
int diameter = dis(a,b);
int ans = diameter;
for(int i = 0; i < n; i++){
if(i==a or i==b) continue;
int branch = (dis(a,i)+dis(b,i)-diameter)/2;
int lpos = dis(a,i)-branch;
int rpos = diameter-lpos;
M[lpos]=max({M[lpos],branch,lpos,rpos});
}
for(auto u : M) ans=min(ans,u.second);
if(sub<=2) return ans;
bool ok = 0;
for(auto u : M){
int lpos = u.first, mx_dis = u.second;
int rpos = diameter-lpos;
if(mx_dis!=ans) continue; ok = 1;
for(int i = 0; i < n; i++)
p[i]=i, sz[i]=1, vis[i]=0;
for(int i = 0; i < n; i++){
if(vis[findSet(i)]) continue;
int i_u = dis(a,i)-lpos;
if(dis(a,i)<dis(b,i)) i_u = dis(b,i)-rpos;
for(int j = i+1; j < n; j++){
int u_j = dis(a,j)-lpos;
if(dis(a,j)<dis(b,j)) u_j = dis(b,j)-rpos;
if(i_u+u_j != dis(i,j)) unionSet(i,j);
}
vis[findSet(i)]=1;
}
for(int i = 0; i < n; i++)
ok&=(sz[findSet(i)]<=n/2);
if(ok) return ans;
}
return -ans;
}
Compilation message
towns.cpp: In function 'int hubDistance(int, int)':
towns.cpp:55:3: warning: this 'if' clause does not guard... [-Wmisleading-indentation]
55 | if(mx_dis!=ans) continue; ok = 1;
| ^~
towns.cpp:55:29: note: ...this statement, but the latter is misleadingly indented as if it were guarded by the 'if'
55 | if(mx_dis!=ans) continue; ok = 1;
| ^~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
344 KB |
Output is correct |
2 |
Correct |
7 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
10 ms |
532 KB |
Output is correct |
5 |
Correct |
10 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
348 KB |
Output is correct |
2 |
Correct |
8 ms |
344 KB |
Output is correct |
3 |
Correct |
10 ms |
348 KB |
Output is correct |
4 |
Correct |
19 ms |
532 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
9 ms |
348 KB |
Output is correct |
2 |
Correct |
10 ms |
536 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
10 ms |
536 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
1 ms |
348 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |