import java.util.*; /* * public class TreeNode { * int val = 0; * TreeNode left = null; * TreeNode right = null; * public TreeNode(int val) { * this.val = val; * } * } */ public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param root1 TreeNode类 * @param root2 TreeNode类 * @return bool布尔型 */ public boolean leafSimilar (TreeNode root1, TreeNode root2) { // write code here List<Integer> list1 = new ArrayList<>(); List<Integer> list2 = new ArrayList<>(); // 定义递归函数dfs,用于遍历二叉树并将叶子节点的值存入列表 dfs(list1, root1); dfs(list2, root2); // 反转列表list2 Collections.reverse(list2); // 判断两个列表是否相等 return list1.equals(list2); } private void dfs(List<Integer> list, TreeNode root) { if (root == null) { return; } if (root.left == null && root.right == null) { list.add(root.val); return; } dfs(list, root.left); dfs(list, root.right); } }
这道题目考察的是二叉树的遍历和列表操作。
- 创建一个类TreeNode,表示二叉树的节点。节点包含一个整数值val以及左子树left和右子树right。
- 创建一个类Solution,包含一个方法leafSimilar,用于判断两棵二叉树的叶子节点是否相同。
- leafSimilar方法接收两个参数root1和root2,分别表示两棵二叉树的根节点。返回一个布尔值,表示两棵树的叶子节点是否相同。
- 在leafSimilar方法中初始化两个列表list1和list2,用于分别存放两棵树的叶子节点的值。
- 定义一个递归函数dfs,用于遍历二叉树并将叶子节点的值存入列表。
- 在dfs函数中,首先判断当前节点是否为空,如果为空则直接返回。
- 然后判断当前节点是否为叶子节点,即左右子树都为空。如果是叶子节点,则将当前节点的值加入到对应的列表中,并返回。
- 如果当前节点不是叶子节点,则递归调用dfs函数遍历左子树和右子树。
- 调用dfs函数,分别传入list1, root1和list2, root2,用于遍历两棵树的叶子节点并将它们存入对应的列表中。
- 反转list2列表,以便后面的比较。
- 判断list1和list2是否相等,如果相等,则返回true,否则返回false。