認識樹 相同的樹

題目

難度：簡單

通過這一題可以很好地了解java中關(guān)于TreeNode的方法，了解樹的概念。

給定兩個二叉樹，編寫一個函數(shù)來檢驗它們是否相同。

如果兩個樹在結(jié)構(gòu)上相同，并且節(jié)點具有相同的值，則認為它們是相同的。

示例 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邏輯表達式，代碼結(jié)構(gòu)更像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)