题目描述
请实现两个函数,分别用来序列化和反序列化二叉树。
思路1:
序列化的方法是不唯一的,如下图,是通过bfs进行二叉树的序列化
bfs序列化二叉树的代码如下:
import java.util.*;
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Solution {
String Serialize(TreeNode root) {
// 方法一 BFS序列化二叉树
if(root == null) {
return "";
}
LinkedList<TreeNode> list = new LinkedList<>(); // 辅助队列
list.add(root);
TreeNode node, flagNode = new TreeNode(-1);
StringBuilder sb = new StringBuilder();
while(!list.isEmpty()) {
int size = list.size();
for(int i = 0; i < size; i++) {
node = list.get(0);
if(node == flagNode) {
sb.append("#").append(","); // 用','隔开每个数字
list.remove(node);
continue;
}
sb.append(node.val).append(",");
if(node.left != null) {
list.add(node.left);
}else {
list.add(flagNode);
}
if(node.right != null) {
list.add(node.right);
}else {
list.add(flagNode);
}
list.remove(node);
}
}
return sb.toString();
}
// 反序列化二叉树
TreeNode Deserialize(String str) {
if("".equals(str)) {
return null;
}
String[] arr = str.split(",");
TreeNode root = new TreeNode(Integer.parseInt(arr[0]));
ArrayList<TreeNode> list = new ArrayList<>(); // 使用辅助ArrayList
list.add(root);
int j = 0;
TreeNode r = list.get(j), node;
for(int i = 0; i < arr.length - 1; i += 2) {
if(!"#".equals(arr[i + 1])) {
node = new TreeNode(Integer.parseInt(arr[i + 1]));
list.add(node);
r.left = node;
} else {
r.left = null;
}
if(!"#".equals(arr[i + 2])) {
node = new TreeNode(Integer.parseInt(arr[i + 2]));
list.add(node);
r.right = node;
} else {
r.right = null;
}
if(++j == list.size()) break;
r = list.get(j);
}
return root;
}
}
思路2:
也可以通过先序遍历,序列化二叉树,遇到空节点,用’#’表示
java 代码如下(反序列化未完成)
static StringBuilder sb = new StringBuilder();
static String Serialize(TreeNode root) { // 序列化
if(root == null) {
sb.append("#,");
return "";
}
sb.append(root.val).append(",");
Serialize(root.left);
Serialize(root.right);
return "";
}
// 反序列化未完成