剑指 Offer 37. 序列化二叉树(困难)
问题描述
请实现两个函数,分别用来序列化和反序列化二叉树。
示例:
你可以将以下二叉树:
1
/ \
2 3
/ \
4 5序列化为 "[1,2,3,null,null,4,5]"
解题思路
首先需要实现serialize方法把二叉树转换为字符串格式,可以看到示例中的序列化结果是层序遍历的结果。因此先使用层序遍历的方法去完成遍历,在遍历的过程中完成字符串的拼接。
另外需要实现deserialize方法把字符串反序列化为二叉树。首选需要把字符串做split拆分操作(以,为间隔),再取第一个元素创建根节点root,同样适用辅助队列去模拟对树的层序遍历。
需要注意的坑:
①需要处理根节点为null的情况,此时如果不在字符串中做处理,反序列化的过程中就会报空指针异常。因此需要在序列化和反序列化的方法中分别对空二叉树做处理。
②在序列化和反序列化的过程中都要处理,左右节点为空的情况。特别是在反序列化中,不能只判断一次当前元素是否是“null”,需要将下标+1之后在重新判断,并且注意,下标的变化必须不管是值为空还是不为空都要+1,否则会影响节点位置。
Java解法
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Codec {
// Encodes a tree to a single string.
public String serialize(TreeNode root) {
if(root==null) return "[]";
Queue<TreeNode> queue = new LinkedList<>();
queue.offer(root);
List<String> res = new ArrayList<>();
while(!queue.isEmpty()){
int size = queue.size();
for(int i=0;i<size;i++){
TreeNode node = queue.poll();
if(node!=null){
res.add(node.val+"");
queue.offer(node.left);
queue.offer(node.right);
}
else res.add("null");
}
}
String str = "[";
for(String s:res){
str+=s;
str+=",";
}
str.substring(0,str.length()-1);
str+="]";
return str;
}
// Decodes your encoded data to tree.
public TreeNode deserialize(String data) {
Queue<TreeNode> queue = new LinkedList<>();
if(data.equals("[]")) return null;
String [] tree = data.substring(1,data.length()-1).split(",");
TreeNode root = new TreeNode(Integer.parseInt(tree[0]));
queue.offer(root);
int ind = 1;
while(!queue.isEmpty()){
TreeNode node = queue.poll();
if(!tree[ind].equals("null")){
TreeNode left = new TreeNode(Integer.parseInt(tree[ind]));
node.left = left;
queue.offer(node.left);
}
ind++;
if(!tree[ind].equals("null")){
TreeNode right = new TreeNode(Integer.parseInt(tree[ind]));
node.right = right;
queue.offer(node.right);
}
ind++;
}
return root;
}
}
// Your Codec object will be instantiated and called as such:
// Codec codec = new Codec();
// codec.deserialize(codec.serialize(root));