首页 > 代码库 > lca(洛谷P3379 最近公共祖先(LCA))
lca(洛谷P3379 最近公共祖先(LCA))
如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先.
输入格式:
第一行包含三个正整数N、M、S,分别表示树的结点个数、询问的个数和树根结点的序号。
接下来N-1行每行包含两个正整数x、y,表示x结点和y结点之间有一条直接连接的边(数据保证可以构成树)。
接下来M行每行包含两个正整数a、b,表示询问a结点和b结点的最近公共祖先。
输出格式:
输出包含M行,每行包含一个正整数,依次为每一个询问的结果.
输入样例#1:
5 5 43 12 45 11 42 43 23 51 24 5
输出样例#1:
44144
思路的话可以见我的转载博文:传送门
这是我的代码实现~
%:pragma GCC optmize(3)//o3优化,尽量不要开 #include<bits/stdc++.h>#define maxn 1100010#define maxm 1100130using namespace std;int n,m,ss;struct aaa{int to,next;}edge[maxn];struct bbb{int same,to,next,num;bool flag;bbb(){flag=0;}}s[maxm];int head[maxn],shead[maxm],pre[maxn],book[maxn],ans[maxm];int cnt=0;inline int read()//注意快读 { char c=getchar(); int num=0; while(!isdigit(c)) c=getchar(); while(isdigit(c)) { num=num*10+c-‘0‘; c=getchar(); } return num;} void add_edge(int u,int v){ edge[++cnt].to=v; edge[cnt].next=head[u]; head[u]=cnt; edge[++cnt].to=u; edge[cnt].next=head[v]; head[v]=cnt;}int cntt=0;void add_search(int u,int v,int w){ s[++cntt].to=v; s[cntt].next=shead[u]; s[cntt].num=w; shead[u]=cntt; s[++cntt].to=u; s[cntt].num=w; s[cntt].next=shead[v]; shead[v]=cntt;}int find(int x){ if(pre[x]==x) return x; else return pre[x]=find(pre[x]);}int merge(int x,int y)//注意谁和谁合并 { int fx=find(x),fy=find(y); if(pre[fy]!=fx) { pre[fy]=fx; return 1; } return 0;}void tarjan(int point,int f){ for(int i=head[point];i;i=edge[i].next) if(edge[i].to!=f&&!book[edge[i].to])//不能返回之前走过的点——父节点 { tarjan(edge[i].to,point); merge(point,edge[i].to); book[edge[i].to]=1; } for(int i=shead[point];i;i=s[i].next) if(book[s[i].to]&&!s[i].flag) { ans[s[i].num]=find(s[i].to); s[i].flag=1;//走过的点标记 int tt=i;if(tt%2)tt++;else tt--; s[tt].flag=1;//因为是双向储存所以两边都要去掉 }}int main(){ scanf("%d%d%d",&n,&m,&ss); for(int i=1;i<n;i++) { int t1,t2; t1=read();t2=read(); add_edge(t1,t2); } for(int i=1;i<=n;i++) pre[i]=i; for(int i=1;i<=m;i++) { int t1,t2; t1=read();t2=read(); add_search(t1,t2,i); } tarjan(ss,0);//虚构一个父节点 for(int i=1;i<=m;i++) printf("%d\n",ans[i]); return 0;}
lca(洛谷P3379 最近公共祖先(LCA))
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。