LintCode 470 [Tweaked Identical Binary Tree]

原题

检查两棵二叉树是否在经过若干次扭转后可以等价。扭转的定义是,交换任意节点的左右子树。等价的定义是,两棵二叉树必须为相同的结构,并且对应位置上的节点的值要相等。
注意:你可以假设二叉树中不会有重复的节点值。

样例

    1             1
   / \           / \
  2   3   and   3   2
 /                   \
4                     4

是扭转后可等价的二叉树。

    1             1
   / \           / \
  2   3   and   3   2
 /             /
4             4

就不是扭转后可以等价的二叉树。

解题思路

  • Recursion - 递归求解,分治的思路。
  • 注意,题目中说的是经过若干次扭转后可以等价,所以不要忘记考虑完全identical的情况,某一个节点的左右子树翻转一次对称,反转两次还原。

完整代码

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        this.val = val
        this.left, this.right = None, None
"""
class Solution:
    """
    @param a, b, the root of binary trees.
    @return true if they are tweaked identical, or false.
    """
    def isTweakedIdentical(self, a, b):
        # Write your code here
        if a is None and b is None:
            return True
        if a and b and a.val == b.val:
            return self.isTweakedIdentical(a.left, b.right) and \
                    self.isTweakedIdentical(a.right, b.left) or \
                    self.isTweakedIdentical(a.left, b.left) and \
                    self.isTweakedIdentical(a.right, b.right)
        return False

你可能感兴趣的:(LintCode 470 [Tweaked Identical Binary Tree])