JZ61 序列化二叉树

序列化二叉树

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

 

二叉树的序列化是指：把一棵二叉树按照某种遍历方式的结果以某种格式保存为字符串，从而使得内存中建立起来的二叉树可以持久保存。序列化可以基于先序、中序、后序、层序的二叉树遍历方式来进行修改，序列化的结果是一个字符串，序列化时通过 某种符号表示空节点（#），以 ！ 表示一个结点值的结束（value!）。

二叉树的反序列化是指：根据某种遍历顺序得到的序列化字符串结果str，重构二叉树。
将二叉树序列化为字符串可以通过前序遍历（Preorder Traversal）来实现

 

func Serialize(root *TreeNode) string {
	if root == nil {
		return "#"
	}

	// 序列化当前节点的值，并递归序列化左右子树
	left := Serialize(root.Left)
	right := Serialize(root.Right)

	// 拼接当前节点值和左右子树序列化结果
	return strconv.Itoa(root.Val) + "," + left + "," + right
}

func Deserialize(data string) *TreeNode {
	nodes := strings.Split(data, ",")
	return buildTree(&nodes)
}

func buildTree(nodes *[]string) *TreeNode {
	if len(*nodes) == 0 {
		return nil
	}

	// 取出当前节点的值
	val := (*nodes)[0]
	*nodes = (*nodes)[1:]

	// 若节点值为 "#"，表示空节点
	if val == "#" {
		return nil
	}

	// 创建当前节点，并递归构建左右子树
	intVal, _ := strconv.Atoi(val)
	node := &TreeNode{Val: intVal}
	node.Left = buildTree(nodes)
	node.Right = buildTree(nodes)

	return node
}