QOJ.ac
QOJ
ID | Problem | Submitter | Result | Time | Memory | Language | File size | Submit time | Judge time |
---|---|---|---|---|---|---|---|---|---|
#201782 | #7338. Education Nightmare | kjhhjki | RE | 1ms | 5776kb | C++14 | 2.2kb | 2023-10-05 16:43:42 | 2023-10-05 16:43:43 |
Judging History
answer
#pragma GCC optimize(2)
#include <bits/stdc++.h>
#define MAXN 200005
#define For(I,A,B) for(int I = (A), endi = (B); I <= endi; ++I)
#define foR(I,A,B) for(int I = (A), endi = (B); I >= endi; --I)
#define ForE(I,A) for(int I = head[A]; I; I = e[I].nxt)
using namespace std;
typedef long long _ll;
typedef unsigned int ui;
struct Edge { int to,nxt; } e[MAXN<<1];
int head[MAXN],tot;
void add_edge(int x,int y) { e[++tot] = (Edge){y,head[x]}; head[x] = tot; }
int T,mx,ans,cnt,dis,pos,n,m,s,u,v,sz,dep[MAXN],pre[MAXN];
bool hasS[MAXN],inS[MAXN];
void dfs(int u, int prev)
{
pre[u] = prev;
if(u == s) hasS[u] = 1;
ForE(i,u)
{
int v = e[i].to;
if(v == prev) continue;
dep[v] = dep[u] + 1;
dfs(v,u);
if(hasS[v]) hasS[u] = 1;
}
sz += hasS[u];
}
void dfs2(int u, int pre)
{
inS[u] = 1;
ForE(i,u)
{
int v = e[i].to;
if(v == pre) continue;
dfs2(v,u);
}
}
int t,a[MAXN];
void solve()
{
cin >> n >> s >> m; sz = mx = ans = 0; int f = 0;
tot = 0; fill(head,head+n+1,0);
fill(hasS,hasS+n+1,0); fill(dep,dep+n+1,0); fill(inS,inS+n+1,0);
For(i,2,n) cin >> u >> v, add_edge(u,v), add_edge(v,u);
dfs(m,0); inS[m] = 1;
ForE(i,m)
if(hasS[e[i].to])
dfs2(e[i].to,m);
For(i,1,n) if(!inS[i]) mx = max(mx,dep[i]), f = 1;
u = s; ans = mx + dep[s];
if(!f) ans = 0x7fffffff;
if(s == m)
{
For(i,1,n) ans = max(ans,dep[i]);
cout << ans << '\n';
return;
}
For(i,1,n) if(inS[i] && !hasS[i]) a[++cnt] = i;
sort(a+1,a+1+cnt,[](int x, int y){ return dep[x] > dep[y]; });
ans = dep[s] + max(mx,dep[a[1]]);
for(pos = 1; pos <= cnt;)
{
u = a[pos];
while(1)
{
if(hasS[u]) break;
hasS[u] = 1; ++sz;
u = pre[u];
}
while(pos <= cnt && hasS[a[pos]]) ++pos;
ans = min(ans,sz*2-2-dep[s]+max(mx,dep[a[pos]]));
}
cout << max(ans,mx) << '\n';
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0); cout.tie(0);
cin >> T;
while(T--) solve();
return 0;
}
Details
Tip: Click on the bar to expand more detailed information
Test #1:
score: 100
Accepted
time: 1ms
memory: 5776kb
input:
1 4 2 3 1 2 1 3 1 4
output:
4
result:
ok 1 number(s): "4"
Test #2:
score: -100
Runtime Error
input:
14966 15 13 8 7 14 7 10 5 3 1 3 15 3 1 13 9 5 7 2 6 13 10 11 13 8 5 10 5 12 14 4 16 14 9 16 11 14 9 8 15 14 15 13 10 6 11 3 2 9 5 6 7 10 6 6 8 1 5 15 4 10 2 11 12 100 49 58 67 43 55 34 84 42 3 74 84 54 20 6 86 83 88 51 2 99 4 78 91 64 14 59 82 38 91 44 24 12 12 2 39 19 43 46 5 80 41 35 80 97 79 8 47...