题目描述:
请实现两个函数,分别用来序列化和反序列化二叉树。
示例:
你可以将以下二叉树:
序列化为 “[1,2,3,null,null,4,5]”
解题思路:
1、层次遍历:
序列化和反序列就是将二叉树转化为一个类似于[1,2,3,null,null,4,5]
的字符串(序列化),然后根据这个字符串生成为原来的二叉树(反序列化)。这里并不需要按照例子中的序列化的结果进行序列化,我们可以按照自己的序列化方式进行序列化,只要满足反序列化后得到的二叉树是正确的即可。
这里我使用层次遍历的方式进行序列化(利用Queue来进行层次遍历),最后序列化的结果为[1,2,3,null,null,4,5,null,null,null,null]
。从这个结果可以看出只要每个节点的左右孩子都进行了序列化,无论该左右孩子是否存在。而反序列化的过程也比较简单,具体过程参见代码。
实现代码:
// Encodes a tree to a single string.
public String serialize(TreeNode root) {
if(root == null)
return "[]";
StringBuilder sb = new StringBuilder("[");
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()){
TreeNode node = queue.poll();
if(node != null){
sb.append(node.val + ",");
queue.add(node.left);
queue.add(node.right);
}
else
sb.append("null,");
}
sb.deleteCharAt(sb.length() - 1);//去掉最后的逗号','
sb.append("]");
return sb.toString();
}
// Decodes your encoded data to tree.
public TreeNode deserialize(String data) {
if(data.equals("[]"))
return null;
String[] vals = data.substring(1,data.length() - 1).split(",");
TreeNode root = new TreeNode(Integer.parseInt(vals[0]));
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
int i = 1;
while (!queue.isEmpty()){
TreeNode node = queue.poll();
if(!vals[i].equals("null")){
node.left = new TreeNode(Integer.parseInt(vals[i]));
queue.add(node.left);
}
i++;
if(!vals[i].equals("null")){
node.right = new TreeNode(Integer.parseInt(vals[i]));
queue.add(node.right);
}
i++;
}
return root;
}