【剑指offer】62.序列化二叉树(python)

题目描述

请实现两个函数，分别用来序列化和反序列化二叉树。

思路

《剑指offer》P283

• 序列化
  序列化的方法可以借鉴先序遍历的思想，当遍历到叶子节点的时候，添加#。
• 反序列化
  反序列化可以通过递归来实现。
  定义一个index记录当前的下标，如果当前是数字，则构造一个TreeNode节点，并递归地向后构造其左右孩子。如果当前是#，则构造一个空结点。返回该结点。

code

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def __init__(self):
        self.index = -1
        
    def Serialize(self, root):
        # write code here
        # 利用先序遍历的思路
        if not root:
            return []
        node = root
        node_stack = []
        prelist = []
        while node or node_stack:
            while node:
                prelist.append(node.val)
                node_stack.append(node)
                node = node.left
            prelist.append('#')
            node = node_stack.pop()
            node = node.right
        # return ','.join(p for p in prelist)
        prelist.append('#')
        return prelist
        
    def Deserialize(self, s):
        # write code here
        self.index += 1
        if self.index >= len(s):
            return None
        node = None
        if s[self.index] != '#':
            node = TreeNode(s[self.index])
            node.left = self.Deserialize(s)
            node.right = self.Deserialize(s)
        return node