叶子相似的树

请考虑一棵二叉树上所有的叶子,这些叶子的值按从左到右的顺序排列形成一个 叶值序列 。

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

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

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

示例 1:

输入:root1 = [3,5,1,6,2,9,8,null,null,7,4], root2 = [3,5,1,6,7,4,2,null,null,null,null,null,null,9,8]
输出:true
示例 2:

输入:root1 = [1,2,3], root2 = [1,3,2]
输出:false

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/leaf-similar-trees
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

/**
 * 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) {
        //分别记录两个根数的叶子
        ArrayList<Integer> list1 = new ArrayList<>();
        ArrayList<Integer> list2 = new ArrayList<>();
        //递归方法遍历树
        recursionRoot(root1,list1);
        recursionRoot(root2,list2);
        return list1.equals(list2);
    }
    public void recursionRoot(TreeNode root , ArrayList list){
        //遍历的节点不存在
        if(root==null)return;
        if(root.left==null&root.right==null){
            list.add(root.val);
            return;
        }
        recursionRoot(root.left,list);
        recursionRoot(root.right,list);
    }


}

迭代器

/**
 * 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 {
    //迭代器写法
    //1.先编写迭代器的接口
    class Iter implements Iterator<TreeNode>{
        TreeNode root;
        Stack<TreeNode> stack;;

        //2.重写迭代器的构造方法和hasNext,next()方法
        Iter(TreeNode root){
            this.root = root;
            this.stack = new Stack<TreeNode>();
            stack.push(root);
        }

        //判断栈中是否为空
        @Override
        public boolean hasNext(){
            return stack.isEmpty();
        }

        @Override
        public TreeNode next(){
            while(true){
                TreeNode curr = stack.pop();
                if(curr.left!=null){
                    stack.push(curr.left);
                }
                if(curr.right!=null){
                    stack.push(curr.right);
                }
                if(curr.left==null&&curr.right==null){
                    return curr;
                }
            }
        }
    }

    public boolean leafSimilar(TreeNode root1, TreeNode root2) {
        //3.判断使用迭代器进行val值判断
        Iter node1 = new Iter(root1);
        Iter node2 = new Iter(root2);
        while(!node1.hasNext()){
            if(node2.hasNext()||node1.next().val!=node2.next().val){
                return false;
            }
        }
        if(!node2.hasNext()){
            return false;
        }else{
            return true;
        }

    }


}
posted @   网抑云黑胶SVIP用户  阅读(10)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 地球OL攻略 —— 某应届生求职总结
· 周边上新:园子的第一款马克杯温暖上架
· Open-Sora 2.0 重磅开源!
· 提示词工程——AI应用必不可少的技术
· .NET周刊【3月第1期 2025-03-02】
点击右上角即可分享
微信分享提示