请实现两个函数,分别用来序列化和反序列化二叉树
思路:序列化就是把二叉树转为数值或字符串序列。注意若子节点为空要入队为“#”。
反序列化则是将字符串转为二叉树,本题是以前序遍历结果转为二叉树。
def Serialize(self, root):
# write code here
if not root:
return '#'
#把根节点入队,再遍历左子树,右子树
return str(root.val) +',' + self.Serialize(root.left) +','+ self.Serialize(root.right)
def Deserialize(self, s):
# write code here
#先把s用分隔符分成数组形式
st=s.split(',')
return self.setTree(st)
def setTree(self,st):
if not st:
return None
#每次数组弹出头部,存入节点,按前序遍历,根左右顺序,用“#”标志左右子树分开
val=st.pop(0)
root=None
if val !='#':
root=TreeNode(int(val))
root.left=self.setTree(st)
root.right=self.setTree(st)
return root