扣第 297 题「二叉树的序列化与反序列化」
就是给你输入一棵二叉树的根节点 root,要求你实现如下一个类:
public class Codec {
// 把一棵二叉树序列化成字符串
public String serialize(TreeNode root) {}
// 把字符串反序列化成二叉树
public TreeNode deserialize(String data) {}
}
我们可以用 serialize 方法将二叉树序列化成字符串,用 deserialize 方法将序列化的字符串反序列化成二叉树,至于以什么格式序列化和反序列化,这个完全由你决定。
比如说输入如下这样一棵二叉树:
serialize 方法也许会把它序列化成字符串 2,1,#,6,3,#,#,其中 # 表示 null 指针,那么把这个字符串再输入 deserialize 方法,依然可以还原出这棵二叉树。也就是说,这两个方法会成对儿使用,你只要保证他俩能够自洽就行了。
前序遍历的实现方式
public class Codec {
String SEQ=",";
String NULL="#";
// 把一棵二叉树序列化成字符串
public String serialize(TreeNode root) {
StringBuilder sb=StringBuilder();
serialize(root,sb);
return sb.toString;
}
//自定义辅助函数,将
public void serialize(TreeNode root,StringBuilder sb){
if(root==null) {
sb.append(NULL).append(SEQ);
return;}
sb.append(root.val).append(SEQ);
serialize(root.left,sb);
serialize(root.right,sb);
}
// 把字符串反序列化成二叉树
public TreeNode deserialize(String data) {
LinkedList<String> nodes=new LinkedList<>();
for(String s:data.split(SEQ){
nodes.add(s);
}
return deserialize(nodes);
}
//利用前序遍历使列表转变为树
public TreeNode deserialize(LinkedList<String> nodes){
if(nodes.isEmpty()) return null;
String first=nodes.poll();
if(first.equals(NULL)) return null;
TreeNode root=new TreeNode(Integer.parseInt(first));
root.left=deserialize(nodes);
root.right=deserialize(nodes);
return root;
}
}