P3379 【模板】最近公共祖先（LCA）

題目描述

如題，給定一棵有根多叉樹，請求出指定兩個點直接最近的公共祖先。

輸入輸出格式

輸入格式：

第一行包含三個正整數N、M、S，分別表示樹的結點個數、詢問的個數和樹根結點的序號。

接下來N-1行每行包含兩個正整數x、y，表示x結點和y結點之間有一條直接連接的邊（數據保證可以構成樹）。

接下來M行每行包含兩個正整數a、b，表示詢問a結點和b結點的最近公共祖先。

輸出格式：

輸出包含M行，每行包含一個正整數，依次為每一個詢問的結果。

輸入輸出樣例

輸入樣例#1：?復制 5 5 4 3 1 2 4 5 1 1 4 2 4 3 2 3 5 1 2 4 5 輸出樣例#1：?復制 4 4 1 4 4

說明

時空限制：1000ms,128M

數據規模：

對于30%的數據：N<=10，M<=10

對于70%的數據：N<=10000，M<=10000

對于100%的數據：N<=500000，M<=500000

樣例說明：

該樹結構如下：

第一次詢問：2、4的最近公共祖先，故為4。

第二次詢問：3、2的最近公共祖先，故為4。

第三次詢問：3、5的最近公共祖先，故為1。

第四次詢問：1、2的最近公共祖先，故為4。

第五次詢問：4、5的最近公共祖先，故為4。

故輸出依次為4、4、1、4、4。

?

#include<cmath> #include<cstdio> #include<cstdlib> #include<iostream> using namespace std;const int M = 500010; int n,m,root,pre[M*4],to[M*4],head[M],num,dep[M],depth; int f[M][25],fa[M],p;void add(int u,int v) {pre[++num]=head[u];to[num]=v;head[u]=num; }void Pre() {p=int(log(n)/log(2)+0.001);for(int i=1; i<=n; i++) if(fa[i]) f[i][0]=fa[i];for(int i=1; i<=p; i++)for(int j=1; j<=n; j++) f[j][i]=f[f[j][i-1]][i-1]; }void dfs(int s) {for(int i=head[s]; i; i=pre[i]) {int v=to[i];if(!dep[v]) {dep[v]=++depth;fa[v]=s;dfs(v);}} }void Ask(int a,int b) {if(dep[a]<dep[b]) swap(a,b);for(int i=p; i>=0; i--) if(dep[f[a][i]]>dep[b])a=f[a][i];printf("%d\n",f[a][0]); }int main() {scanf("%d%d%d",&n,&m,&root);for(int i=1; i<n; i++) {int x,y;scanf("%d%d",&x,&y);add(x,y);add(y,x);}depth=1;dep[root]=1;dfs(root);Pre();for(int i=1; i<=m; i++) {int a,b;scanf("%d%d",&a,&b);Ask(a,b);}return 0; } 板子

?

轉載于:https://www.cnblogs.com/wsdestdq/p/7800914.html

總結

以上是生活随笔為你收集整理的LCA模板的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。