题目描述
如下图所示的一棵二叉树的深度、宽度及结点间距离分别为:
深度:4 宽度:4(同一层最多结点个数)
结点间距离: ⑧→⑥为8 (3×2+2=8)
⑥→⑦为3 (1×2+1=3)
注:结点间距离的定义:由结点向根方向(上行方向)时的边数×2,
与由根向叶结点方向(下行方向)时的边数之和。
输入格式
输入文件第一行为一个整数n(1≤n≤100),表示二叉树结点个数。接下来的n-1行,表示从结点x到结点y(约定根结点为1),最后一行两个整数u、v,表示求从结点u到结点v的距离。
输出格式
三个数,每个数占一行,依次表示给定二叉树的深度、宽度及结点u到结点v间距离。
输入输出样例
输入 #1复制
10 1 2 1 3 2 4 2 5 3 6 3 7 5 8 5 9 6 10 8 6
输出 #1复制
4 4 8
经过剖析样例,我微微思索手动模拟后发现 要找最近的公共祖先,一次次地向上找就好了嘛,其实这有点并查集地意思
寻找时可分为两种情况
1.两点在不同子树中,有公共祖先
2.其中一个点是另一个的祖先
那么代码可以这么写
谁在下面谁就向上走,u向上时用depp1记录,v向上时depp2记录
#include<cstdio> #include<cmath> #include<iostream> using namespace std; int next[100040]; int to[100040]; int st[1000040]; int de[100040]; int dn[100040]; int ru[100040]; int fa[100040][50]; int n,k,tot=0; void add(int x,int y) { to[++tot]=y; next[tot]=st[x]; st[x]=tot; } void build(int num,int fath) { de[num]=de[fath]+1; dn[de[num]-1]++; fa[num][0]=fath; for(int i=1;i<=29;i++) fa[num][i]=fa[fa[num][i-1]][i-1]; for(int i=st[num];i;i=next[i]) if(to[i]!=fath) build(to[i],num); } int LCA(int x,int y) { if(de[x]<de[y]) { int u=x; x=y; y=u; } for(int i=30;i>=0;i--) { if(de[fa[x][i]]>=de[y]) x=fa[x][i]; if(x==y) return x; } for(int i=30;i>=0;i--) if(fa[x][i]!=fa[y][i]) { x=fa[x][i]; y=fa[y][i]; } return fa[x][0]; } int main() { scanf("%d",&n); for(int i=1;i<=n-1;i++) { int x,y; scanf("%d%d",&x,&y); add(y,x); add(x,y); } build(1,0); int ma=0; for(int i=1;i<=n;i++) if(de[i]>ma) ma=de[i]; printf("%d\n",ma); int lo=0; for(int i=1;i<=ma;i++) if(dn[i]>lo) lo=dn[i]; printf("%d\n",lo); int x,y; scanf("%d%d",&x,&y); printf("%d\n",2*(de[x]-de[LCA(x,y)])+de[y]-de[LCA(x,y)]); return 0; }