题目描述
请实现两个函数,分别用来序列化和反序列化二叉树
二叉树的序列化是指:把一棵二叉树按照某种遍历方式的结果以某种格式保存为字符串,从而使得内存中建立起来的二叉树可以持久保存。序列化可以基于先序、中序、后序、层序的二叉树遍历方式来进行修改,序列化的结果是一个字符串,序列化时通过 某种符号表示空节点(#),以 ! 表示一个结点值的结束(value!)。
二叉树的反序列化是指:根据某种遍历顺序得到的序列化字符串结果str,重构二叉树。
思路
递归
方法一是照着大佬方法写的,不是通常的先序遍历序列。
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
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
if not s:
return None
news=s.split(',')
root,index=self.Deserialize_new(news,0)
return root
def Deserialize_new(self,s,index):
if index>=len(s) or s[index]=='#':
return None,index+1
node=TreeNode(int(s[index]))
index+=1
node.left,index=self.Deserialize_new(s,index)
node.right,index=self.Deserialize_new(s,index)
return node,index
方法二
层次遍历序列化,按通常的先序遍历执行反序列化。
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def Serialize(self, root):
# write code here
if not root:
return "#"
#层次遍历存
res=[]
queue=[]
queue.append(root)
while queue:
flag=False
for node in queue:
if node:
flag=True
break
if flag!=True:
return res
nextline=[]
for node in queue:
if node:
res.append(node.val)
nextline.append(node.left)
nextline.append(node.right)
else:
res.append('#')
nextline.append(None)
nextline.append(None)
queue=nextline
return res
def Deserialize(self, s):
if not s:
return None
root=self.Deserialize_new(s,0)
return root
def Deserialize_new(self,s,index):
if index>=len(s):
return None
if s[index]=='#':
return None
node=TreeNode(int(s[index]))
node.left=self.Deserialize_new(s,index*2+1)
node.right=self.Deserialize_new(s,index*2+2)
return node