题目
请实现两个函数,分别用来序列化和反序列化二叉树。
你需要设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
代码实现
层序遍历
public class Codec {
// Encodes a tree to a single string.
public String serialize(TreeNode root) {
if (root == null){
return "[]";
}
StringBuilder builder = new StringBuilder("[");
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()){
TreeNode node = queue.poll();
if (node != null){
builder.append(node.val+",");
queue.offer(node.left);
queue.offer(node.right);
}else {
builder.append("null,");
}
}
//删掉最后一个,号
builder.deleteCharAt(builder.length()-1);
builder.append(']');
System.out.println(builder.toString());
return builder.toString();
}
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<TreeNode>() {{ 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;
}
}