字节跳动:寻找二叉树两节点的最近祖节点

题目描述

给定一棵二叉树以及这棵树上的两个节点 o1 和 o2,请找到 o1 和 o2 的最近公共祖先节点。

解:

这道题可以采用递归查找+标记法求解,先找到两个节点的位置并用栈保存路径,(因为没有父节点指向没办法回滚)。然后对路径进行重合检查即可。

这道题有点特殊,返回值只能是int,也就是说不存在找不到的情况。所以边界检查可做可不做。

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 * }
 */

public class Solution {
    /**
     * 
     * @param root TreeNode类 
     * @param o1 int整型 
     * @param o2 int整型 
     * @return int整型
     */
    Stack<Integer> s1=new Stack<Integer>();
    Stack<Integer> s2=new Stack<Integer>();
    
    public boolean find(TreeNode ro,int o,Stack s){
        if(ro==null)
            return false;
        s.push(ro.val);
        if(ro.val==o){
            return true;
        }
        if(find(ro.left,o,s)){
            return true;
        }
        if(find(ro.right,o,s)){
            return true;
        }
        s.pop();
        return false;
    }
    
    public int lowestCommonAncestor (TreeNode root, int o1, int o2) {
        int v1=0,v2=0;
        if(!find(root,o1,s1)){
            return 0;
        }
        //因为递归单次查找速度较快,这里就做两次查找了,保持代码简洁
        if(!find(root,o2,s2)){
            return 0;
        }
        //size()即树的层次,从1开始计算,最近祖节点层<=min(o1层,o2层)
        while(s1.size()>s2.size()){
            s1.pop();
        }
        while(s2.size()>s1.size()){
            s2.pop();
        }
        while(!s1.empty() && !s2.empty()){
            v1=s1.pop();
            v2=s2.pop();
            if(v1==v2){
                return v1;
            }
        }
        return 0;
    }
}

 

©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页