LeetCode - 100. Same Tree(判断两棵树是否完全相同)(简单题)

最新推荐文章于 2022-01-21 12:03:27 发布

zxzxin

最新推荐文章于 2022-01-21 12:03:27 发布

阅读量186

点赞数

分类专栏：树

本文链接：https://blog.csdn.net/zxzxzx0119/article/details/83239353

版权

树专栏收录该内容

33 篇文章 0 订阅

订阅专栏

LeetCode - 100. Same Tree(判断两棵树是否完全相同)(简单题)

递归
非递归

题目链接

题目

在这里插入图片描述

递归

这题比较简单。分为五种情况:

p、q都为空，返回true；
p == null && q != null返回false；
p != null && q == null返回false；
p.val != q.val 返回false；
p.val == q.val —> 去判断他们的左右子树是否相等(递归)；

class Solution {
    public boolean isSameTree(TreeNode p, TreeNode q) {
        if (p == null && q == null)
            return true;
        if (p == null || q == null)
            return false;
        return p.val == q.val && isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
    }
}

非递归

这个也很简单:

栈中保存两棵树的节点即可(使用数组)，当然也可以使用两个栈；
然后利用前序非递归类似的方式非递归，判断条件还是和递归一样，只不过改成了非递归而已。

class Solution {
    public boolean isSameTree(TreeNode p, TreeNode q) {
        if (p == null && q == null)
            return true;
        if (p == null || q == null)
            return false;
        Stack<TreeNode[]> stack = new Stack<>();
        stack.push(new TreeNode[]{p, q});
        while (!stack.isEmpty()) {
            TreeNode[] top = stack.pop();
            if (top[0] == null && top[1] == null)
                continue;
            if (top[0] == null || top[1] == null)
                return false;
            if (top[0].val != top[1].val)
                return false;
            stack.push(new TreeNode[]{top[0].left, top[1].left});
            stack.push(new TreeNode[]{top[0].right, top[1].right});
        }
        return true;
    }
}