LeetCode Same Tree

最新推荐文章于 2020-07-27 10:14:22 发布

原创最新推荐文章于 2020-07-27 10:14:22 发布 · 498 阅读

0 ·

CC 4.0 BY-SA版权

OJ 同时被 2 个专栏收录

986 篇文章

订阅专栏

算法设计与分析

321 篇文章

订阅专栏

本文介绍了一种判断两个二叉树是否相等的方法，提供了递归和非递归两种实现方式。递归方法通过深度优先搜索比对每个节点的值及子树来判断；非递归方法则使用队列模拟递归过程，逐层比对节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Given two binary trees, write a function to check if they are equal or not.

Two binary trees are considered equal if they are structurally identical and the nodes have the same value.

题意：给出两个二叉树，判断这两个树是不是相等

思路：用DFS，首先比较当前结点的值是不是相等，然后比较左子树，再比较右子树

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public boolean isSameTree(TreeNode p, TreeNode q) {
        if (p == null && q == null) return true;
        else if (p == null && q != null) return false;
        else if (p != null && q == null) return false;
        else {
            if (p.val != q.val) return false;
            
            if (!isSameTree(p.left, q.left) || !isSameTree(p.right, q.right)) return false;
            
            return true;
        }
    }
}

第二种方法：用非递归方法，模拟递归的出栈、入栈

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public boolean isSameTree(TreeNode p, TreeNode q) {
        Queue<TreeNode> q1 = new LinkedList<TreeNode>();
        Queue<TreeNode> q2 = new LinkedList<TreeNode>();
        
        q1.add(p);
        q2.add(q);
        
        while (!q1.isEmpty() && !q2.isEmpty()) {
            TreeNode node1 = q1.poll();
            TreeNode node2 = q2.poll();
            if (node1 == null && node2 == null) continue;
            
            if (node1 == null || node2 == null) return false;
            
            if (node1.val != node2.val) return false;
            
            q1.add(node1.left); q2.add(node2.left);
            q1.add(node1.right); q2.add(node2.right);
        }
        
        return true;
    }
}