认识树 相同的树

2020-06-16 15:05 更新

题目

难度:简单

通过这一题可以很好地了解java中关于TreeNode的方法,了解树的概念。

给定两个二叉树,编写一个函数来检验它们是否相同。

如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。

示例 1:

输入:       1         1
          / \       / \
         2   3     2   3


        [1,2,3],   [1,2,3]


输出: true

示例 2:

输入:      1          1
          /           \
         2             2


        [1,2],     [1,null,2]


输出: false

示例 3:

输入:       1         1
          / \       / \
         2   1     1   2


        [1,2,1],   [1,1,2]


输出: false

解法一:递归

/*
**by wulara
*/


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

解法二:Python

利用 and/or 简化python逻辑表达式,代码结构更像pythoner

## Definition for a binary tree node.
## class TreeNode(object):
##     def __init__(self, x):
##         self.val = x
##         self.left = None
##         self.right = None


class Solution(object):
    def isSameTree(self, p, q):
        """
        :type p: TreeNode
        :type q: TreeNode
        :rtype: bool
        """
        def dfs(p, q, flag):
            if flag:
                if not p or not q:
                    flag = not p and not q or False
                else:
                    flag = p.val == q.val and dfs(p.left, q.left, flag) and dfs(p.right, q.right, flag)
            return flag
        return dfs(p, q, True)



以上内容是否对您有帮助:
在线笔记
App下载
App下载

扫描二维码

下载编程狮App

公众号
微信公众号

编程狮公众号