324 最近公共祖先 倍增算法
作者:互联网
视频链接:
// P3379 【模板】最近公共祖先(LCA) #include <iostream> #include <cstring> #include <algorithm> #include <vector> using namespace std; const int N=5e5+10; int n,m,s,a,b; vector<int> e[N]; int dep[N],fa[N][20]; void dfs(int u, int father){ dep[u]=dep[father]+1; // 向上跳1,2,4...步的祖先节点 fa[u][0]=father; for(int i=1; i<=19; i++) fa[u][i]=fa[fa[u][i-1]][i-1]; for(int v : e[u]) if(v!=father) dfs(v, u); } int lca(int u, int v){ if(dep[u]<dep[v])swap(u, v); // 先跳到同一层 for(int i=19; i>=0; i--) if(dep[fa[u][i]]>=dep[v]) u=fa[u][i]; if(u==v) return v; // 再跳到LCA的下一层 for(int i=19; i>=0; i--) if(fa[u][i]!=fa[v][i]) u=fa[u][i], v=fa[v][i]; return fa[u][0]; } int main(){ scanf("%d%d%d", &n,&m,&s); for(int i=1; i<n; i++){ scanf("%d%d",&a,&b); e[a].push_back(b); e[b].push_back(a); } dfs(s, 0); while(m--){ scanf("%d%d", &a, &b); printf("%d\n",lca(a, b)); } return 0; }
标签:dep,include,int,d%,father,fa,算法,324,倍增 来源: https://www.cnblogs.com/dx123/p/16320461.html