设计一个算法,并编写代码来序列化和反序列化二叉树。将树写入一个文件被称为“序列化”,读取文件后重建同样的二叉树被称为“反序列化”。
如何反序列化或序列化二叉树是没有限制的,你只需要确保可以将二叉树序列化为一个字符串,并且可以将字符串反序列化为原来的树结构。
给出一个测试数据样例, 二叉树{3,9,20,#,#,15,7},表示如下的树结构:
3
/ \
9 20
/ \
15 7
数据是按照BFS遍历得到的,BFS遍历也叫广度优先搜索,以上面的为例子,就是一层一层地遍历,第一层,遍历得到3,第二层遍历得到9和10,第三层遍历得到#,#,15,7。
下面是实现的代码:
/**
* Definition of TreeNode:
* public class TreeNode {
* public int val;//节点的值
* public TreeNode left, right;//左右节点
* public TreeNode(int val) {
* this.val = val;//初始化节点
* this.left = this.right = null;//将左右节点置为0
* }
* }
*/
class Solution {
/**
* This method will be invoked first, you should design your own algorithm
* to serialize a binary tree which denote by a root node to a string which
* can be easily deserialized by your own "deserialize" method later.
*/
public static Queue<TreeNode> queue = new LinkedList<>();//用于保存将树结构序列化成字符串时使用来保存
public static Queue<TreeNode> dataqueue = new LinkedList<>();//用于将字符串结构反序列化成树时使用来保存节点
public String string = "";
public String serialize(TreeNode root) {
// write your code here
if (root == null) {//如果根节点为空,整棵树为空,返回“#”
return "#";
} else {
queue.add(root);、、根节点不为空,加入队列
while (!queue.isEmpty()) {//直到队列为空,才跳出循环
TreeNode treeNode = queue.poll();//每次取出队列的第一的节点
if (treeNode == null)//如果为空,那么直接输出“#”,不再把该点的左右子节点加入队列
string = string + "#" + ",";
else {//该节点不为空,则把该点的值加上在字符串后面
string = string + treeNode.val + ",";
queue.add(treeNode.left);//该节点不为空,那么它的左右子节点都要加进去,可能为空,可能不为空,加进去不判断
queue.add(treeNode.right);//从队列中取出来才判断是不是为空,为空则没有下一层了
}
}
}
string = string.substring(0, string.length() - 2);//将后面的“,”去掉
return string;
}
/**
* This method will be invoked second, the argument data is what exactly you
* serialized at method "serialize", that means the data is not given by
* system, it's given by your own serialize method. So the format of data is
* designed by yourself, and deserialize it here as you serialize it in
* "serialize" method.
*/
public TreeNode deserialize(String data) {
// write your code here
String[] strs = data.split(",");//根据“,”来分隔成为字符串数组
if (strs.length == 0||data.equals("#")) {//判断字符串数组是等于#或者为空
TreeNode rooTreeNode = null;
return rooTreeNode;//直接返回空根节点
} else {//数组中至少有一个元素
TreeNode rootNode = new TreeNode(Integer.parseInt(strs[0]));//拿到数组的第一个元素
dataqueue.add(rootNode);//把根节点加进去队列中(思路是把已经创建好的节点加进去队列中)
int i = 1;下次取出第二个元素
while (!dataqueue.isEmpty() && i < strs.length) {//直到队列为空,或者数组中元素被取完了
TreeNode treeNode = dataqueue.poll();//每次弹出队列的第一个节点,分别构造左右节点
if (strs[i].equals("#")) {//如果等于“#”,那么取出来的这个节点的左节点为空
treeNode.left = null;
i++;
} else {//如果不是空节点,那么我们要把这个节点加入队列中,这意味着这个节点可能还有子节点
treeNode.left = new TreeNode(Integer.parseInt(strs[i]));
dataqueue.add(treeNode.left);
i++;
}//下面接着判断右节点,注意,有可能数组被取光了,也就是没有右节点了
if (i < strs.length&&strs[i].equals("#") ) {
treeNode.right = null;
i++;
} else if(i < strs.length&& !strs[i].equals("#")){
treeNode.right = new TreeNode(Integer.parseInt(strs[i]));
dataqueue.add(treeNode.right);
i++;
}
}
return rootNode;
}
}
}//在上面,最主要的是使用了队列的数据结构,然后把第一个节点加进去(不为空),再每次从队列中取出节点,直到队列中的节点被取完,当然,在循环中,如果符合条件(节点不为空,
//就会往队列中加入节点),这样就实现了,同一层的都是挨着加入队列的,出来的时候,也是紧挨着的,而不是不断递归到最底层,优先忽略了同一层的其他节点。
如果有所帮助,脸皮厚求个赞~
此文章仅代表自己(本菜鸟)学习积累记录,或者学习笔记,如有侵权,请联系作者删除。人无完人,文章也一样,文笔稚嫩,在下不才,勿喷,如果有错误之处,还望指出,感激不尽~
技术之路不在一时,山高水长,纵使缓慢,驰而不息。
公众号:秦怀杂货店