ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

洛谷 P3379 【模板】最近公共祖先(LCA)题解(树剖做法)

2021-08-22 10:02:56  阅读:144  来源: 互联网

标签:结点 洛谷 树剖 int 题解 top cnt 1000001 siz


题目描述

如题,给定一棵有根多叉树,请求出指定两个点直接最近的公共祖先。

输入格式

第一行包含三个正整数 N,M,S 分别表示树的结点个数、询问的个数和树根结点的序号。

接下来 N−1 行每行包含两个正整数 x,y,表示 x 结点和 y 结点之间有一条直接连接的边(数据保证可以构成树)。

接下来 M 行每行包含两个正整数 a,b,表示询问 a 结点和 b 结点的最近公共祖先。

输出格式

输出包含 M 行,每行包含一个正整数,依次为每一个询问的结果。

AC代码(树剖)

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 
 4 int head[1000001],to[1000001],next[1000001],cnt;
 5 int n,m,s,x,y,maxn;
 6 int fa[1000001],top[1000001],dep[1000001],siz[1000001],zs[1000001];
 7 
 8 void add(int u,int v)
 9 {
10     cnt++;
11     next[cnt]=head[u];
12     head[u]=cnt;
13     to[cnt]=v;
14 }
15 
16 int init(int k,int f,int d)
17 {
18     fa[k]=f;
19     dep[k]=d;
20     siz[k]=1;
21     maxn=-1;
22     for (int i=head[k];i;i=next[i])
23     {
24         if (to[i]!=f)
25         {
26             siz[k]+=init(to[i],k,d+1);
27             if (siz[to[i]]>maxn)
28             {
29                 maxn=siz[to[i]];
30                 zs[k]=to[i];
31             }
32         }
33     }
34     return siz[k];
35 }
36 
37 void dfs(int k,int Top)
38 {
39     top[k]=Top;
40     if (zs[k]==0)
41         return ;
42     dfs(zs[k],Top);
43     for (int i=head[k];i;i=next[i])
44     {
45         if (top[to[i]]==0)
46             dfs(to[i],to[i]);
47     }
48 }
49 
50 int query(int x,int y)
51 {
52     while (top[x]!=top[y])
53     {
54         if (dep[top[x]]<dep[top[y]])
55             swap(x,y);
56         x=fa[top[x]];
57     }
58     if (dep[x]<dep[y])
59         return x;
60     else
61         return y;
62 }
63 
64 int main()
65 {
66     scanf("%d%d%d",&n,&m,&s);
67     for (int i=1;i<n;i++)
68     {
69         scanf("%d%d",&x,&y);
70         add(x,y); add(y,x);
71     }
72     init(s,0,1);
73     dfs(s,s);
74     while (m--)
75     {
76         scanf("%d%d",&x,&y);
77         printf("%d\n",query(x,y));
78     }
79     return 0;
80 }

 

标签:结点,洛谷,树剖,int,题解,top,cnt,1000001,siz
来源: https://www.cnblogs.com/zeng-xin/p/15171330.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有