设计一个算法,并编写代码来序列化和反序列化二叉树。将树写入一个文件被称为“序列化”,读取文件后重建同样的二叉树被称为“反序列化”。
如何反序列化或序列化二叉树是没有限制的,你只需要确保可以将二叉树序列化为一个字符串,并且可以将字符串反序列化为原来的树结构。
样例
给出一个测试数据样例, 二叉树{3,9,20,#,#,15,7}
,表示如下的树结构:
3
/ \
9 20
/ \
15 7
我们的数据是进行BFS遍历得到的。当你测试结果wrong answer时,你可以作为输入调试你的代码。
你可以采用其他的方法进行序列化和反序列化。
/**
* 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;
* }
* }
*/
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;
}
}
}