一、题目

请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。
872. 叶子相似的树-每日一题 - 图1

举个例子,如上图所示,给定一棵叶值序列为 (6, 7, 4, 9, 8) 的树。

如果有两棵二叉树的叶值序列是相同,那么我们就认为它们是 叶相似 的。

如果给定的两个根结点分别为 root1 和 root2 的树是叶相似的,则返回 true;否则返回 false 。

点击查看原题

二、思路

需要对叶子节点从左到右依次记录其数值,如果记录完全一致,那么就是叶相似,数量或者数值不一致,那就不是叶相似。
比较经典且容易实现的,就是使用中序遍历的方式,中序遍历保证了可以依次从左往右访问叶子节点,使用两个List记录遍历过的节点,进行对比List的长度和每个元素数值,就可以实现本题要求。

三、代码

  1. class Solution {
  2. public boolean leafSimilar(TreeNode root1, TreeNode root2) {
  3. List<Integer> memo1 = new ArrayList();
  4. List<Integer> memo2 = new ArrayList();
  5. inorder(root1, memo1);
  6. inorder(root2, memo2);
  7. if (memo1.size() != memo2.size()) { // 如果叶子节点数量不一致,不是叶相似
  8. return false;
  9. }
  10. for (int i = 0; i < memo1.size(); i++) {
  11. if (memo1.get(i) != memo2.get(i)) { // 有任何一个元素不相等,则不是叶相似
  12. return false;
  13. }
  14. }
  15. return true;
  16. }
  17. // 中序遍历从左往右依次访问叶子节点
  18. public void inorder(TreeNode root, List<Integer> memo) {
  19. if (root == null) {
  20. return ;
  21. }
  22. inorder(root.left, memo); // 先访问左孩子
  23. if (root.left == null && root.right == null) { // 当叶子时记录值
  24. memo.add(root.val);
  25. }
  26. inorder(root.right, memo);
  27. }
  28. }

m、n分别为两棵树的节点规模,时间复杂度为O(m+n),空间复杂度为O(m+n)。