#include <bits/stdc++.h>
#define forn(i,j,n) for(ll i = (ll)j;i<=(ll)n;i++)
#define nfor(i,j,n) for(ll i = (ll)j;i>=(ll)n;i--)
using namespace std;
typedef long long ll;
typedef pair<ll,ll> pii;
const ll N = 2000001,LOG = 17;
ll n,t,m,dep[N],d[N],src;
vector<pair<ll,pii> > g[N];
ll dfs1(ll u,ll prnt,ll depth){
dep[u] = depth;
d[u] = depth;
ll ret = 0;
forn(i,0,g[u].size() - 1){
ll v = g[u][i].second.second;
if(v == prnt)
continue;
ll tmp = dfs1(v,u,depth+1);
ret = max(ret,tmp);
dep[u] = max(dep[u],dep[v]);
g[u][i].first = tmp;
g[u][i].second.first = -dep[v];
}
if(u == t)
return 1;
return ret;
}
void dfs(ll node,ll prnt,ll &one,ll &two,ll in){
if(node == t)
return;
ll id = 0;
while(one <= two && id < g[node].size() && g[node][id].first != 1){
if(g[node][id].second.second != prnt && g[node][id].first != -1)
one++;
g[node][id].first = -1;
id++;
}
if(in == 1)
src = node;
id = 0;
while(id < g[node].size() && g[node][id].first != 1){
if(g[node][id].second.second != prnt && g[node][id].first != -1){
++two;
dfs(g[node][id].second.second,node,one,two,2);
}
id++;
}
if(in != 1){
++one;
++two;
return;
}
while(id < g[node].size()){
if(g[node][id].second.second != prnt && g[node][id].first == 1){
++two;
dfs(g[node][id].second.second,node,one,two,1);
}
id++;
}
}
int main(){
scanf("%lld%lld%lld",&n,&t,&m);
forn(i,1,n-1){
ll u,v;
scanf("%lld%lld",&u,&v);
g[u].push_back(make_pair(0,make_pair(0,v)));
g[v].push_back(make_pair(0,make_pair(0,u)));
}
dfs1(m,0,0);
forn(i,1,n){
sort(g[i].begin(),g[i].end());
}
ll one = 0,two = 0;
dfs(m,0,one,two,1);
cout<<one;
return 0;
}
Compilation message
mousetrap.cpp: In function 'void dfs(ll, ll, ll&, ll&, ll)':
mousetrap.cpp:34:25: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
while(one <= two && id < g[node].size() && g[node][id].first != 1){
~~~^~~~~~~~~~~~~~~~
mousetrap.cpp:43:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
while(id < g[node].size() && g[node][id].first != 1){
~~~^~~~~~~~~~~~~~~~
mousetrap.cpp:55:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
while(id < g[node].size()){
~~~^~~~~~~~~~~~~~~~
mousetrap.cpp: In function 'int main()':
mousetrap.cpp:65:7: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%lld%lld%lld",&n,&t,&m);
~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~
mousetrap.cpp:68:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%lld%lld",&u,&v);
~~~~~^~~~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
52 ms |
47352 KB |
Output is correct |
2 |
Correct |
47 ms |
47452 KB |
Output is correct |
3 |
Correct |
51 ms |
47540 KB |
Output is correct |
4 |
Correct |
47 ms |
47540 KB |
Output is correct |
5 |
Correct |
43 ms |
47540 KB |
Output is correct |
6 |
Correct |
52 ms |
47540 KB |
Output is correct |
7 |
Correct |
43 ms |
47540 KB |
Output is correct |
8 |
Correct |
44 ms |
47540 KB |
Output is correct |
9 |
Correct |
48 ms |
47540 KB |
Output is correct |
10 |
Correct |
44 ms |
47540 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
662 ms |
133692 KB |
Output is correct |
2 |
Correct |
653 ms |
133692 KB |
Output is correct |
3 |
Incorrect |
1439 ms |
133692 KB |
Output isn't correct |
4 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
52 ms |
47352 KB |
Output is correct |
2 |
Correct |
47 ms |
47452 KB |
Output is correct |
3 |
Correct |
51 ms |
47540 KB |
Output is correct |
4 |
Correct |
47 ms |
47540 KB |
Output is correct |
5 |
Correct |
43 ms |
47540 KB |
Output is correct |
6 |
Correct |
52 ms |
47540 KB |
Output is correct |
7 |
Correct |
43 ms |
47540 KB |
Output is correct |
8 |
Correct |
44 ms |
47540 KB |
Output is correct |
9 |
Correct |
48 ms |
47540 KB |
Output is correct |
10 |
Correct |
44 ms |
47540 KB |
Output is correct |
11 |
Incorrect |
46 ms |
133692 KB |
Output isn't correct |
12 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
52 ms |
47352 KB |
Output is correct |
2 |
Correct |
47 ms |
47452 KB |
Output is correct |
3 |
Correct |
51 ms |
47540 KB |
Output is correct |
4 |
Correct |
47 ms |
47540 KB |
Output is correct |
5 |
Correct |
43 ms |
47540 KB |
Output is correct |
6 |
Correct |
52 ms |
47540 KB |
Output is correct |
7 |
Correct |
43 ms |
47540 KB |
Output is correct |
8 |
Correct |
44 ms |
47540 KB |
Output is correct |
9 |
Correct |
48 ms |
47540 KB |
Output is correct |
10 |
Correct |
44 ms |
47540 KB |
Output is correct |
11 |
Correct |
662 ms |
133692 KB |
Output is correct |
12 |
Correct |
653 ms |
133692 KB |
Output is correct |
13 |
Incorrect |
1439 ms |
133692 KB |
Output isn't correct |
14 |
Halted |
0 ms |
0 KB |
- |