算法----- 给定一颗二叉树,找到二叉树上任意两个节点之间的距离(Java版本)

题目:

 给定一颗二叉树,找到二叉树上任意两个节点之间的距离
class TreeNode {
        TreeNode left;
        TreeNode right;
    }

思路:

首先找到一个节点的路径,然后找到另一个节点的路径,针对这两个路径进行去重。把这两个路径的长度相加就可以得到结果。

解决方法:

class TreeNode {
        TreeNode left;
        TreeNode right;
    }

    int findDistance(TreeNode root, TreeNode p, TreeNode q){
        if (p == q) {
            return 0;
        }
        List<TreeNode> list1= findPath(root,p);
        List<TreeNode> list2 = findPath(root,q);
//        TreeNode lastSame = null;
        int lastSame = 0;
        for(int i = 0; i< list1.size();i++){
            if(list1.get(i) == list2.get(i)){
                lastSame = i;
            }else{
                break;
            }
        }

        return list1.size() - lastSame + list2.size() - lastSame;
    }

    List<TreeNode> findPath(TreeNode root,TreeNode node){
        List<TreeNode> list = new ArrayList<>();
        if(root == null){
            return list;
        }

        if(root == node){
            list.add(root);
            return list;
        }
        List<TreeNode> left = findPath(root,root.left);
        List<TreeNode> right = findPath(root,root.right);
        if(left.contains(node)){
            list.addAll(left);
        }else if(right.contains(node)){
            list.addAll(right);
        }
        return list;
    }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值