LeetCode 101.对称二叉树 python/scala

2020-12-30  本文已影响0人  电饭锅娃儿

Symmetric Tree

环境:python 3.6,scala 2.11.8

题意

判断是否对称二叉树(以根节点为中垂线镜像对称,包括结构和节点值)。

分析

转换思路:先判断两颗树是否相同(结构+节点值),再判断根节点的左子树和右子树是否相同即可;

这里只写出最简写法,更多写法参考相同的树

代码

python

def isSymmetric(root):
    """
    :type root: TreeNode
    :rtype: bool
    """
    def dfs(p, q):
        if not p and not q: return True
        if not p or not q: return False
        left_same = dfs(p.left, q.right)
        right_same = dfs(p.right, q.left)
        return p.val == q.val and left_same and right_same
    return dfs(root.left, root.right) if root else True

scala

object LC101 {
  def isSymmetric(root: TreeNode): Boolean = {
    def dfs(p: TreeNode, q: TreeNode): Boolean = (p, q) match {
      case (null, null) => true
      case (null, _) | (_, null) => false
      case _ => p.value == q.value & dfs(p.left, q.right) & dfs(p.right, q.left)
    }
    if (root == null) true else dfs(root.left, root.right)
  }
}

最后

欢迎交流和补充

上一篇下一篇

猜你喜欢

热点阅读