题解 | #在二叉树中找到两个节点的最近公共祖先#

在二叉树中找到两个节点的最近公共祖先

http://www.nowcoder.com/practice/e0cc33a83afe4530bcec46eba3325116

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整型
     */
    boolean flag = false;
    public int lowestCommonAncestor (TreeNode root, int o1, int o2) {
        // write code here
        List<Integer> list1 = new ArrayList<>();
        List<Integer> list2 = new ArrayList<>();
        findPath(root,o1,list1);
        flag = false;
        findPath(root,o2,list2);
        int i = 0;
        int val = 0;
        while(i<Math.min(list1.size(),list2.size())){
            int path1 = list1.get(i);
            int path2 = list2.get(i);
            i++;
            if(path1 == path2){
                val = path1;
            }else{
                break;
            }
        }
        return val;
    }
    //回溯
    public void findPath(TreeNode root, int o,List<Integer> path){
        if(root == null){
            return;
        }
        path.add(root.val);
        int size = path.size();
        if(root.val == o){
            flag = true;
            return;
        }
        findPath(root.left,o,path);
        if(flag){
            return;
        }
        findPath(root.right,o,path);
        if(flag){
            return;
        }
        path.remove(size - 1);
    }
}
全部评论

相关推荐

陆续:不可思议 竟然没那就话 那就我来吧 :你是我在牛客见到的最美的女孩
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务