序列化二叉树
更多题解,欢送关注公众号【程序员学长】,欢送退出
问题形容
剑指 Offer 37. 序列化二叉树
请实现两个函数,别离用来序列化和反序列化二叉树。你须要设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只须要保障一个二叉树能够被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
示例:
输出:root = [1,2,3,null,null,4,5]
输入:[1,2,3,null,null,4,5]
剖析问题
咱们都晓得对于不同的二叉树进行先序、中序、后序或者层序遍历失去的遍历后果有可能是雷同的,即如果晓得了遍历序列,咱们不能确定惟一的一颗二叉树,而题目要求的序列化和反序列化是一对可逆操作,即对一棵树进行序列化后,咱们还能通过序列化后的后果反序列化还原成原来的那棵树。所以咱们须要对二叉树的遍历过程革新一下,这里以层序遍历为例。在遍历的过程中,为了残缺的示意二叉树,思考将叶节点下的 null 也记录。上面咱们来看一下具体的实现。
序列化
咱们能够借助队列来实现二叉树的层序遍历,在层序遍历的过程中,咱们把越过叶子节点的null也退出序列化的后果中。具体算法流程如下所示:
- 如果给定的二叉树为空,间接返回"[]"
- 初始化一个队列queue,并把根节点退出。
对二叉树进行层序遍历。当队列为空时跳出。
- 节点出队,记为node。
- 如果node不为空,将node.val退出后果列表中,并且将node的左、右子节点入队。如果node为空,将“null”退出后果列表中。
- 拼接后果列表,用
','
隔开,首尾增加中括号;并将其返回。
反序列化
利用队列按层构建二叉树。具体算法流程如下所示:
- 如果data为空,间接返回null。
- 初始化列表vals(去掉首尾括号,再用逗号隔开),指针 i = 1 ,将根节点 root (值为 vals[0] )入队;
按层构建二叉树,当队列为空时跳出;
- 节点出队,记为node。
- 构建 node 的左子节点,node.left 的值为 vals[i] ,并将 node.left 入队;而后执行i=i+1。
- 构建 node 的右子节点,node.right 的值为 vals[i] ,并将 node.right 入队;而后执行i=i+1。
- 返回根节点 root 即可。
上面咱们来看一下代码的实现。
class Codec: def serialize(self, root): #如果二叉树为空,间接返回"[]" if not root: return "[]" #初始化一个队列 queue = collections.deque() #将根节点退出队列中 queue.append(root) #后果列表 res = [] #记录null的个数,遇到非空节点就将之前的null节点退出到后果列表中 null_count=0 while queue: node = queue.popleft() if node: #遇到非空节点就将之前的null节点退出到后果列表中 if null_count != 0: res.extend(["null"] * null_count) null_count=0 res.append(str(node.val)) #node的左、右子节点入队 queue.append(node.left) queue.append(node.right) else: #记录null节点的个数 null_count=null_count+1 return '[' + ','.join(res) + ']' def deserialize(self, data): #如果data为"[]",代表二叉树是空,间接返回 if data == "[]": return #初始化 vals = data[1:-1].split(',') i=1 #初始化根节点,并将其入队 root = TreeNode(int(vals[0])) queue = collections.deque() queue.append(root) while queue: if i>=len(vals): break node = queue.popleft() #增加左子节点 if vals[i] != "null": node.left = TreeNode(int(vals[i])) queue.append(node.left) i += 1 if i>=len(vals): break #增加右子节点 if vals[i] != "null": node.right = TreeNode(int(vals[i])) queue.append(node.right) i += 1 return root