二叉树的序列化与反序列化

2020-11-20 14:45:32 浏览数 (1)

代码语言:javascript复制
序列化是将一个数据结构或者对象转换为连续的比特位的操作,进而可以将转换后的数据存储在一个文件或者内存中,同时也可以通过网络传输到另一个计算机环境,采取相反方式重构得到原数据。

请设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。

示例: 

你可以将以下二叉树:

    1
   / 
  2   3
     / 
    4   5

序列化为 "[1,2,3,null,null,4,5]"
提示: 这与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。

说明: 不要使用类的成员 / 全局 / 静态变量来存储状态,你的序列化和反序列化算法应该是无状态的。

这道题的在leecode里面标记是困难,但是实现起来我感觉比较简单,可能是没怎么要求效率 ,有个问题是恢复的时候需要记录现在恢复到第几层了,用了个status去记录。

代码语言:javascript复制
package main

import (
    "fmt"
    "strconv"
    "strings"
)

func main() {
    root := &TreeNode{
        Val: 1,
        Left: &TreeNode{
            Val: 2,
        },
        Right: &TreeNode{
            Val: 3,
            Left: &TreeNode{
                Val: 4,
            },
            Right: &TreeNode{
                Val: 5,
            },
        },
    }
    code := Constructor()
    str := code.serialize(root)
    fmt.Println(str)
    fmt.Printf("% vn",code.deserialize(str))
}

type TreeNode struct {
    Val   int
    Left  *TreeNode
    Right *TreeNode
}

type Codec struct {
    status []string
}

func Constructor() Codec {
    return Codec{}
}

// Serializes a tree to a single string.
func (this *Codec) serialize(root *TreeNode) string {
    if root == nil {
        return ""
    }
    return strings.Trim(this.getString(root), ",")
}

func (this *Codec) getString(root *TreeNode) string {
    if root == nil {
        return ",null"
    }
    str := ","   strconv.Itoa(root.Val)
    str  = this.getString(root.Left)
    str  = this.getString(root.Right)
    return str
}

// Deserializes your encoded data to tree.
func (this *Codec) deserialize(data string) *TreeNode {
    if len(data) <= 0 {
        return nil
    }
    datas := strings.Split(data, ",")
    this.status = datas
    return this.getNode()
}

func (this *Codec) getNode() *TreeNode {
    if len(this.status) <= 0 {
        return nil
    }
    if strings.Compare(this.status[0], "null") == 0 {
        this.status = this.status[1:]
        return nil
    }
    val, err := strconv.Atoi(this.status[0])
    if err != nil {
        return nil
    }
    this.status = this.status[1:]
    tree := &TreeNode{
        Val:   val,
        Left:  this.getNode(),
        Right: this.getNode(),
    }
    return tree

}

0 人点赞