LeetCode 872. 叶子相似的树

最新推荐文章于 2024-09-12 19:02:20 发布

远不及你的微笑

最新推荐文章于 2024-09-12 19:02:20 发布

阅读量304

点赞数 2

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/weixin_55057111/article/details/137483357

版权

算法专栏收录该内容

174 篇文章 0 订阅

订阅专栏

文章介绍了如何使用深度优先搜索算法在两个给定的二叉树（TreeNode）中，分别存储根节点的叶子节点值，然后对比两个列表的元素，判断它们是否相似。

摘要由CSDN通过智能技术生成

解题思路

    来一次深度优先搜素，把数据放到s1中，然后复制一份到s2中，
清空s1,再来一次深度优先搜索，把数据放到s1中，然后将s1和s2进
行比对。

相关代码

/**
 * 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 {
    List<Integer> s1 = new ArrayList<>();
    List<Integer> s2 = new ArrayList<>();
    public boolean leafSimilar(TreeNode root1, TreeNode root2) {
        dfs(root1);
        for(int i=0;i<s1.size();i++){
            int temp = s1.get(i);
            s2.add(temp);
        }
        s1.clear();
        dfs(root2);
        if(s2.size()!=s1.size()) return false;
        for(int i=0;i<s2.size();i++){
            if(s1.get(i)!=s2.get(i)){
                return false;
            }
        }
        return true;
    }

    public void dfs(TreeNode root){
        if(root==null) return;
        if(root.left==null&&root.right==null){
            s1.add(root.val);
        }
        
        dfs(root.left);
        dfs(root.right);
    }
}