温馨提示×

温馨提示×

您好,登录后才能下订单哦!

密码登录×
登录注册×
其他方式登录
点击 登录注册 即表示同意《亿速云用户服务条款》

刷题系列 - 序列化和反序列化一个二叉树

发布时间:2020-08-12 13:46:36 来源:ITPUB博客 阅读:130 作者:张国平 栏目:编程语言

序列化和反序列化一个二叉树,是很开放的一题,就是给出一个二叉树,用序列化方法生成一个字符串;然后用反序列化方法把这个字符串生成原来二叉树。这个在编程时候各个类型一般都有序列化的,用于存储。

这里面要用到python中list转化字符串方法 ','.join(list), 和字符串转换为list的方法string.split(',')。

其实可以用之前刷题的几个题目来组合,比如遍历二叉树生成中序和后序两个队列,合并为一个队列,作为序列化方法;然后有一题是按照中序和后序队列生成二叉树,就可以作为反序列化的方法使用。当然,这样会有很多冗余数据。

其实这个题目比较麻烦的地方就是优化,实现倒是很不难。

我这边用了序列化层级遍历,就是从根节点到叶子节点一层层按照从左到用遍历,如果某个节点的左或者右子节点为空,用#号代替;最后叶子节点下面会都是”#“号,这里做了个判断,如果某层都是#号,视作为空,结束遍历。

反序列化采用对应的方法,这里不多说,看代码即可。

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Codec:
    def serialize(self, root):
        """Encodes a tree to a single string.
        
        :type root: TreeNode
        :rtype: str
        """
        if root != None:
            checkList = [root]
        else:
            checkList = []
        AllNodeList = []
        while checkList != []:
            nextList = []
            for Node in checkList:
                if Node != '#':
                    AllNodeList.append(str(Node.val))
                    if Node.left == None:
                        nextList.append('#')
                    else:
                        nextList.append(Node.left)
                    if Node.right == None:
                        nextList.append('#')
                    else:
                        nextList.append(Node.right)
                else:
                    AllNodeList.append(Node)
            if len(set(nextList)) == 1 and '#' in nextList:
                nextList = []
            checkList = nextList
        return ','.join(AllNodeList)
        
    def deserialize(self, data):
        """Decodes your encoded data to tree.
        
        :type data: str
        :rtype: TreeNode
        """
        if data == '':
            currentLevel = []
            root = None
        else:
            AllNodeList = data.split(",")
            root = TreeNode(int(AllNodeList.pop(0)))
            currentLevel =[root]
        while currentLevel != [] and AllNodeList!= []:
            nextLevel = []
            for node in currentLevel:
                leftValue = AllNodeList.pop(0)
                if leftValue != '#':
                    node.left = TreeNode(int(leftValue))
                    nextLevel.append(node.left)
                    
                rightValue = AllNodeList.pop(0)
                if rightValue != '#':
                    node.right = TreeNode(int(rightValue))
                    nextLevel.append(node.right)
            print([node.val for node in nextLevel])
            currentLevel = nextLevel
        return root  
# Your Codec object will be instantiated and called as such:
# codec = Codec()
# codec.deserialize(codec.serialize(root))
向AI问一下细节

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

AI