Leetcode 剑指 Offer 37. 序列化二叉树

Leetcode 剑指 Offer 37. 序列化二叉树

题目

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

示例:

你可以将以下二叉树：

    1
   / \
  2   3
     / \
    4   5

序列化为 "[1,2,3,null,null,4,5]"

思路

• 根据题目的要求来， 我们可以将题目的示例的二叉树序列化 -> “[1,2,3,null,null,4,5,null,null,null,null]”
• 先讲序列化的过程：如果是空， 那么返回"[]", 如果不是空， 那么借助层次遍历， 空节点记录成null， 节点之间的分隔符采用’,’
• 再讲反序列化的过程： 如果是"[]", 那么直接返回None，如果不是"[]", 说明这棵树不是空树， 反序列化的方法就是再次借助层次遍历和一个下标变量idx， 以题目的示例为例子， 1的左右分别是2和3， 2的左右分别是null和null， 3的左右分别是4和5， 发现到没， 2， 3， null， null， 4， 5刚好是顺序的， 下标变量idx就是记录这个用的， 当然了，空节点不需要进队列， 因为空节点也没有左右节点可构建

代码 —— python

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Codec:
    from collections import deque

    def serialize(self, root):
        """Encodes a tree to a single string.
        
        :type root: TreeNode
        :rtype: str
        """
        if not root: return "[]"

        queue, datalist = deque([root]), []

        while queue:
            node = queue.popleft()
            
            if not node: datalist.append("null")
            else: datalist.append(str(node.val))

            if node: 
                queue.append(node.left)
                queue.append(node.right)      

        return "[" + ','.join(datalist) + "]" 

    def deserialize(self, data):
        """Decodes your encoded data to tree.
        
        :type data: str
        :rtype: TreeNode
        """
        if data == "[]": return None

        datalist = data[1:-1].split(',')
        root = TreeNode(int(datalist[0]))
        queue, idx = deque([root]), 1

        while queue:
            node = queue.popleft()

            if datalist[idx] != "null":
                node.left = TreeNode(int(datalist[idx]))
                queue.append(node.left)

            idx += 1

            if datalist[idx] != "null":
                node.right = TreeNode(int(datalist[idx]))
                queue.append(node.right)

            idx += 1

        return root
        

# Your Codec object will be instantiated and called as such:
# codec = Codec()
# codec.deserialize(codec.serialize(root))