二叉树中任意两个节点之间的最短路径
程序员文章站
2024-03-15 12:18:59
...
思路:两个节点之间的最短路径一定会经过两个节点的最小公共祖先,所以我们可以用LCA(最低公共祖先)的解法。不同于LCA的是,我们返回不只是标记,而要返回从目标结点递归回当前节点的路径。当遇到最小公共祖先的时候便合并路径。需要注意的是,我们要单独处理目标节点自身是最小公共祖先的情况。
Java代码实现如下:
public LinkedList<TreeNode> helper(TreeNode n, TreeNode p, TreeNode q){
if(n == null){
return null;
}
LinkedList<TreeNode> left = helper(n.left, p, q);
LinkedList<TreeNode> right = helper(n.right, p, q);
// 当左右都为空时
if(left == null && right == null){
// 如果当前节点是目标节点,开启一条新路径
if(n == p || n == q){
LinkedList l = new LinkedList<TreeNode>();
l.add(n);
return l;
} else {
// 否则标记为空
return null;
}
// 如果左右节点都不为空,说明是最小公共祖先节点,合并两条路径
} else if(left != null && right != null){
finalPath.addAll(left);
finalPath.add(n);
Collections.reverse(right);
finalPath.addAll(right);
return left;
// 如果当前节点是目标结点,且某一个子树不为空时,说明最小公共祖先是节点自身
} else if (left != null){
left.add(n);
if(n == p || n == q){
finalPath.addAll(left);
}
return left;
} else {
right.add(n);
if(n == p || n == q){
finalPath.addAll(right);
}
return right;
}
}
上一篇: 求100以内所有质数
下一篇: Java 如何编写泛型方法
推荐阅读
-
计算二叉树任意两个节点之间的最短路径长度(Java)
-
二叉树中任意两个节点之间的最短路径
-
算法----- 给定一颗二叉树,找到二叉树上任意两个节点之间的距离(Java版本)
-
求二叉树两节点的LCA(最近公共祖先)(必须要会+常考)+求叶子节点到LCA的距离+求两个叶子节点之间的距离
-
剑指offer之求二叉树中两个节点的最低共同父节点
-
LeetCode 题目-671. 二叉树中第二小的节点/687.最长同值路径(python实现)
-
求二叉树中两个节点的最近公共祖先(三叉链,搜索树,普通二叉树)
-
判断一棵树是否是完全二叉树和求二叉树中两个节点的最近公共祖先——题集(十三)
-
给定两个非空二叉树 s 和 t,检验s 中是否包含和 t 具有相同结构和节点值的子树
-
二叉树中,打印根节点到指定节点的路径——后序遍历的变型解答