-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfind_lca
131 lines (113 loc) · 2.63 KB
/
find_lca
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
//manish kumar patel
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define ld long double
#define rep(i,a,b) for(ll i=a;i<b;i++)
#define repb(i,a,b) for(ll i=a;i>=b;i--)
#define err() cout<<"--------------------------"<<endl;
#define errA(A) for(auto i:A) cout<<i<<" ";cout<<endl;
#define err1(a) cout<<#a<<" "<<a<<endl
#define err2(a,b) cout<<#a<<" "<<a<<" "<<#b<<" "<<b<<endl
#define err3(a,b,c) cout<<#a<<" "<<a<<" "<<#b<<" "<<b<<" "<<#c<<" "<<c<<endl
#define err4(a,b,c,d) cout<<#a<<" "<<a<<" "<<#b<<" "<<b<<" "<<#c<<" "<<c<<" "<<#d<<" "<<d<<endl
#define pb push_back
#define all(A) A.begin(),A.end()
#define allr(A) A.rbegin(),A.rend()
#define ft first
#define sd second
#define pii pair<int,int>
#define pll pair<ll,ll>
#define V vector<ll>
#define S set<ll>
#define VV vector<V>
#define Vpll vector<pll>
#define endl "\n"
#define LOG 20
VV dp;
V depth;
void pre_comp(VV &adj,int n,int root,int par)
{
// err2(root,par);
depth[root]=depth[par]+1;
dp[root][0]=par;
rep(i,1,LOG)
{
dp[root][i]=dp[dp[root][i-1]][i-1];
}
for(auto it:adj[root])
{
if(it !=par)
pre_comp(adj,n,it,root);
}
}
int lca(int x,int y,int n)
{
if(depth[x]<depth[y])
{
swap(x,y);
}
int kk=depth[x]-depth[y];
rep(k,0,LOG)
{
if(kk&(1<<k))
{
x=dp[x][k];
}
}
// err1(x);
if(x==y)
{
return x;
}
repb(i,LOG-1,0)
{
if(dp[x][i]!=dp[y][i])
{
x=dp[x][i];
y=dp[y][i];
}
}
return dp[x][0];
}
int main()
{
ios_base::sync_with_stdio(false);
cin.tie(NULL); cout.tie(NULL);
#ifndef ONLINE_JUDGE
clock_t tStart = clock();
freopen("D://input.txt","r",stdin);
freopen("D://output.txt","w",stdout);
#endif
int t;
// cin>>t;
t=1;
while(t--)
{
ll n,q,root;
cin>>n>>q;
root=1;
// err2(n,q);
dp.resize(n+1,V(LOG,0));
depth.resize(n+1,0);
VV adj(n+1);
rep(i,0,n-1)
{
int x,y;
cin>>x>>y;
adj[x].pb(y);
adj[y].pb(x);
// err2(x,y);
}
pre_comp(adj,n,root,0);
rep(i,0,q)
{
int x,y;
cin>>x>>y;
cout<<lca(x,y,n)<<"\n";
}
}
#ifndef ONLINE_JUDGE
printf("\nRun Time -> %.10fs\n", (double)(clock()-tStart) / CLOCKS_PER_SEC);
#endif
}