刚学树链剖分,提交全是RE,样例是对的,求调。
#include<bits/stdc++.h>
using namespace std;
const int N=500505;
int m,n,s;
vector <int> g[N];
struct node{
int father;//父亲
int dep;//深度
int size;//子树上的节点个数(包括自己)
int son;//重儿子
int top;//重链顶部节点
int dfn;//dfs序
}a[N];
int rnk[N];
int dfs1(int u){
a[u].son=-1;
a[u].size=1;
int i,j;
for(i=0;i<g[u].size();i++){
int v=g[u][i];
if(a[u].father!=v){
a[v].dep=a[u].dep+1;
a[v].father=u;
dfs1(v);
a[u].size=a[u].size+a[v].size;
if(a[u].son==-1||a[v].size>a[a[u].son].size)a[u].son=v;
}
}
}
int cnt=0;
void dfs2(int u,int t){
a[u].top=t;
cnt++;
a[u].dfn=cnt;
rnk[cnt]=u;
if(a[u].son==-1)return ;
dfs2(a[u].son,t);
int i;
for(i=0;i<g[u].size();i++){
int v=g[u][i];
if(v!=a[u].father&&v!=a[u].son){
dfs2(v,v);
}
}
}
int lca(int u,int v){
while(a[u].top!=a[v].top){
if(a[a[u].top].dep>=a[a[v].top].dep){
u=a[a[u].top].father;
}else{
v=a[a[v].top].father;
}
}
if(a[u].dep<a[v].dep)return u;
else return v;
}
int main(){
cin>>n>>m>>s;
// a[s].dep=1;
int u,v;
for(int i=1;i<n;i++){
cin>>u>>v;
g[u].push_back(v);
g[v].push_back(u);
}
a[s].dep=0;
dfs1(s);
dfs2(s,s);
// for(int i=1;i<=n;i++){
// cout<<a[i].top<<" ";
// }
for(int i=0;i<m;i++){
cin>>u>>v;
printf("%d\n",lca(u,v));
}
}