Python对称的二叉树
分为递归和非递归的两种方式,思想是一样的。主要就是把叶子节点的None节点也加入到遍历当中。按照前序遍历二叉树,存入一个序列中。然后按照和前序遍历对应的先父节点,然后右子节点,最后左子节点遍历二叉树,存入一个序列。如果前后两个序列相等,那么说明二叉树是对称的。
- '''
- 请实现一个函数,用来判断一颗二叉树是不是对称的。
- 注意,如果一个二叉树同此二叉树的镜像是同样的,定义其为对称的。
- '''
-
- # -*- coding:utf-8 -*-
- class TreeNode:
- def __init__(self, x):
- self.val = x
- self.left = None
- self.right = None
- class Solution:
- def isSymmetrical(self, pRoot):
- return self.selfIsSymmetrical(pRoot, pRoot)
- def selfIsSymmetrical(self, pRoot1, pRoot2):
- if pRoot1 == None and pRoot2 == None:
- return True
- if pRoot1 == None or pRoot2 == None:
- return False
- if pRoot1.val != pRoot2.val:
- return False
- return self.selfIsSymmetrical(pRoot1.left, pRoot2.right) and self.selfIsSymmetrical(pRoot1.right, pRoot2.left)
- # 非递归实现判断二叉树是否对称
- # 利用前序遍历
- class Solution2:
- def isSymmetrical(self, pRoot):
- preList = self.preOrder(pRoot)
- mirrorList = self.mirrorPreOrder(pRoot)
- if preList == mirrorList:
- return True
- return False
-
- def preOrder(self, pRoot):
- if pRoot == None:
- return [None]
- treeStack = []
- output = []
- pNode = pRoot
- while pNode or len(treeStack) > 0:
- while pNode:
- treeStack.append(pNode)
- output.append(pNode.val)
- pNode = pNode.left
- if not pNode:
- output.append(None)
- if len(treeStack):
- pNode = treeStack.pop()
- pNode = pNode.right
- if not pNode:
- output.append(None)
- return output
-
- def mirrorPreOrder(self, pRoot):
- if pRoot == None:
- return [None]
- treeStack = []
- output = []
- pNode = pRoot
- while pNode or len(treeStack) > 0:
- while pNode:
- treeStack.append(pNode)
- output.append(pNode.val)
- pNode = pNode.right
- if not pNode:
- output.append(None)
- if len(treeStack):
- pNode = treeStack.pop()
- pNode = pNode.left
- if not pNode:
- output.append(None)
- return output
-
- pNode1 = TreeNode(8)
- pNode2 = TreeNode(6)
- pNode3 = TreeNode(10)
- pNode4 = TreeNode(5)
- pNode5 = TreeNode(7)
- pNode6 = TreeNode(9)
- pNode7 = TreeNode(11)
-
- pNode1.left = pNode2
- pNode1.right = pNode3
- pNode2.left = pNode4
- pNode2.right = pNode5
- pNode3.left = pNode6
- pNode3.right = pNode7
-
- S = Solution2()
- result = S.isSymmetrical(pNode1)
- print(result)
解压密码: detechn或detechn.com
免责声明
本站所有资源出自互联网收集整理,本站不参与制作,如果侵犯了您的合法权益,请联系本站我们会及时删除。
本站发布资源来源于互联网,可能存在水印或者引流等信息,请用户自行鉴别,做一个有主见和判断力的用户。
本站资源仅供研究、学习交流之用,若使用商业用途,请购买正版授权,否则产生的一切后果将由下载用户自行承担。