首页 > 代码库 > 最近公共祖先(9018_1892)
最近公共祖先(9018_1892)
这道题是LCA裸题
我只是练模板的
#include<iostream> #include<cstdio> #include<algorithm> #include<cstring> #include<vector> #include<queue> using namespace std; inline int read(){ int t=1,num=0;char c=getchar(); while(c>‘9‘||c<‘0‘){if(c==‘-‘)t=-1;c=getchar();} while(c>=‘0‘&&c<=‘9‘){num=num*10+c-‘0‘;c=getchar();} return num*t; } const int maxn=500010; vector<int>g[maxn]; int d[maxn],n,m,ro,f[maxn][21]; void bfs(int s){ memset(d,127/3,sizeof(d)); queue<int>q;q.push(s);d[s]=0; while(!q.empty()){ int x=q.front();q.pop(); for(int i=0;i<g[x].size();i++){ int y=g[x][i]; if(d[y]>d[x]+1){ d[y]=d[x]+1;q.push(y); } } } } void bz(){ for(int j=1;j<=20;j++) for(int i=1;i<=n;i++) f[i][j]=f[f[i][j-1]][j-1]; } int lca(int x,int y){ if(d[x]<d[y])swap(x,y); int tmp=d[x]-d[y]; for(int i=0;i<=20;i++) if((1<<i)&tmp)x=f[x][i]; if(x==y)return x; for(int i=20;i>=0;i--) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i]; return f[x][0]; } int main() { n=read();m=read(); for(int i=1;i<n;i++){ int a=read(),b=read(); g[a].push_back(b);f[b][0]=a; if(!f[a][0])ro=a; } bfs(ro);bz(); for(int i=1;i<=m;i++){ int a=read(),b=read(); printf("%d\n",lca(a,b)); } return 0; }
本文由Yzyet编写,网址为www.cnblogs.com/Yzyet。非Yzyet同意,禁止转载,侵权者必究。
最近公共祖先(9018_1892)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。