LeetCode 872. 叶子相似的树

文章目录

  • 题目
  • 思路
  • 代码实现(Java)
  • 思考
  • 总结


题目

LeetCode 872. 叶子相似的树_第1张图片
LeetCode 872. 叶子相似的树_第2张图片
LeetCode 872. 叶子相似的树_第3张图片


思路

先序遍历两个二叉树得到两个树的叶值序列,比较即可。


代码实现(Java)

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
     
    public boolean leafSimilar(TreeNode root1, TreeNode root2) {
     
        List<Integer> r1 = new LinkedList<Integer>();
        List<Integer> r2 = new LinkedList<Integer>();
        dfs(root1,r1);
        dfs(root2,r2);
        if(r1.size() != r2.size()){
     
            return false;
        }
        for(int i = 0; i<r1.size(); i++) {
     
            if(r1.get(i).intValue() != r2.get(i).intValue()){
     
                return false;
            }
        }
        return true;
    }

    public void dfs(TreeNode root,List<Integer> list){
     
        if(root!=null) {
     
            if(root.left == null && root.right == null) {
     
                list.add(root.val);
            }
            dfs(root.left,list);
            dfs(root.right,list);
        }
    }
}

思考

List 可以用 equals 进行比较,但是时间复杂度较高。 学习 List 中的 equals 方法源码如下

public boolean equals(Object o) {
         
   if (o == this)        
      return true;    
   // 判断是否是List列表,只要实现了List接口就是List列表
   if (!(o instanceof List))        
      return false;    
   // 遍历list所有元素
   ListIterator<E> e1 = listIterator();
   ListIterator e2 = ((List) o).listIterator();    
   while (e1.hasNext() && e2.hasNext()) {
     
       E o1 = e1.next();
       Object o2 = e2.next();        
       // 有不相等的就退出
       if (!(o1==null ? o2==null : o1.equals(o2)))            
           return false;
    }    
    // 长度是否相等
    return !(e1.hasNext() || e2.hasNext());
}

总结

每天都最好看看一些经常用到的API的源码,并且要自己思考源码的各方面优缺点。带着疑问刷题学习效率比较高。


坚持分享,坚持原创,喜欢博主的靓仔靓女们可以看看博主的首页博客!
您的点赞与收藏是我分享博客的最大赞赏!
博主博客地址: https://blog.csdn.net/weixin_43967679

你可能感兴趣的:(LeetCode刷题记录,leetcode,java,dfs)