一、题目
链接:来源:牛客网
有一棵无穷大的满二叉树,其结点按根结点一层一层地从左往右依次编号,根结点编号为1。现在有两个结点a,b。请设计一个算法,求出a和b点的最近公共祖先的编号。
给定两个int a,b。为给定结点的编号。请返回a和b的最近公共祖先的编号。注意这里结点本身也可认为是其祖先。
二、思路
这道题我之前是做过的,我确实也做出来了,但是写的代码真是又臭又长。当时写完脑子就懵了,也不清楚怎么写出这种垃圾代码。
过了一天,仔细想了想,整理了下思路。
根据题目设定,可以知道。满二叉树的子节点与父节点之间的关系为root = child / 2 。利用这个关系,如果a != b,就让其中的较大数除以2, 如此循环知道a == b, 即是原来两个数的近公共祖先。
三、代码
及其不推荐的代码,效率低,不易懂(我都看不懂)
public class LCA {
public int getLCA(int a, int b) {
// 如果其中有一个是根节点,直接返回1,就没有计算的必要了
if (a == 1 || b == 1) {
return 1;
}
int min = Math.min(a, b);
int max = Math.max(a, b);
// 使得max的值小于等于min
while (max > min) {
max /= 2;
}
// 如果max == min,就说明找到了公共祖先结点,即min是祖先结点
if (min == max) {
return min;
}
// 如果上面的不成立,max 的值小于 min 且min不是祖先结点,此时有两种情况
// 1.此时两个结点在同一层(且在同一子树下)
if (max == min / 2) {
return max;
}
// 2.此时两个结点不在同一层(且不在同一子树下)
if (max / 2 == min / 2) {
return min / 2;
}
return getLCA(max / 2, min / 2);
}
}
改进后的简易写法
public class LCA {
public int getLCA(int a,int b){
while (a != b){
if (a > b){
a /= 2;
}else {
b /= 2;
}
}
return a;
}
}