判断两颗二叉树是否相等
给定两颗二叉树根节点 root1 和 root2,编写一个函数检验它们是否相同。如果两棵树再结构上相同,并且相同位置上的值也相等,则认为他们是相等的
这种题,直接对两棵树进行递归,中间如果结构不相同或者值不相同就返回false,一直判断到叶子节点,到叶子节点都没问题就返回true,代码如下
//判断两棵树是否相同
public static boolean isSameTree(Tree tree1, Tree tree2) {
if (tree1 == null & tree2 == null) {
return true;
}
if (tree1 == null) {
return false;
}
if (tree2 == null) {
return false;
}
if (tree1.data != tree2.data) {
return false;
}
return isSameTree(tree1.leftTree, tree2.leftTree) &&
isSameTree(tree1.rightTree, tree2.rightTree);
}
判断一棵二叉树是否对称
给定一棵二叉树,检查它是否镜像对称,示例如下
不对称的如下:
这道题,其实和上一题差不多。先拿到根节点判断是否为空,然后取出根节点的左右子树,题目的镜像对称,换成算法表示,就是左子树的左边等于右子树的右边,左子树的右边等于右子树的左边,本质上还是相等判断。代码如下:
//对称二叉树,检查一棵树是否对称
public static boolean isSymmetry(Tree root) {
if (root == null) {
return true;
}
return check(root.leftTree, root.rightTree);
}
public static boolean check(Tree tree1, Tree tree2) {
if (tree1 == null && tree2 == null) {
return true;
}
if (tree1 == null || tree2 == null) {
return false;
}
if (tree1.data != tree2.data) {
return false;
}
return check(tree1.leftTree, tree2.rightTree) &&
check(tree1.rightTree, tree2.leftTree);
}
二叉树所有路径
给定一棵二叉树根节点root,按任意顺序,返回所有根节点到叶子结点的路径
这题乍一看无从下手,先拆解一下,理一下思路:
- 有多少个叶子节点,就有多少条路径
- 可以用递归遍历到每一个叶子节点,到了叶子结点时再加入结果数组中
可是,到了叶子节点只能拿到叶子节点的数据,拿到前面节点的数据呢?一个小技巧:每一次遍历时,将当前节点的数据都存到一个String里,作为参数传给下一个递归调用。这样最后到了叶子节点,就有所有的数据了。
代码如下
//写出二叉树所有路径
public static List<String> binaryTreePath(Tree root) {
List<String> res = new ArrayList<>();
//根节点为空,直接返回
if(root == null){
return res;
}
//不为空,将根节点加到字符串,开始遍历
dfs(root,String.valueOf(root.data),res);
return res;
}
public static void dfs(Tree root, String path, List<String> res) {
//到了叶子节点,将整条路径加入res
if (root.leftTree == null && root.rightTree == null) {
res.add(path);
return;
}
//没到叶子节点,判断左右子树是否为空,不为空就加入路径
if (root.leftTree != null) {
dfs(root.leftTree, path + "->" + root.leftTree.data, res);
}
if (root.rightTree != null) {
dfs(root.rightTree, path + "->" + root.rightTree.data, res);
}
}