问题描述:
请设计一个算法来实现二叉树的序列化与反序列化。需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。
思路:
- 序列化:将二叉树序列化为字符串。显然可以通过深度优先搜索,选择前序遍历来一次序列化二叉树,当子树为null时到达递归临界条件返回。注意用“ ,”分割开每个树对应的节点值。在反序列化中才可以将每个节点值从字符串中分割识别出来。
public class SerializeMethod{
public String serialize(TreeNode root) {
return rserialize(root, "");
}
public String rserialize(TreeNode root, String str) {
if (root == null) {
str += "None,";
} else {
str += str.valueOf(root.val) + ",";
str = rserialize(root.left, str);
str = rserialize(root.right, str);
}
return str;
}
- 反序列化:将字符串反序列化为二叉树。反序列的逻辑和序列化时是要等价的,因此同样采用先序遍历的逻辑,首先将字符串通过“ ,”分隔后转换成链表dataList。通过调用递归函数,每次递归读取链表的第一个值,若为None,说明该节点为null,作为临界条件。若不为空,则创建节点并赋节点值,然后从字符链表中删除该节点代表的值,调用递归分别查找左右子树实现先序遍历。最后返回根节点
public TreeNode deserializeMethod(String data) {
String[] dataArray = data.split(",");
List<String> dataList = new LinkedList<String>(Arrays.asList(dataArray));
return rdeserialize(dataList);
}
public TreeNode rdeserialize(List<String> dataList) {
if (dataList.get(0).equals("None")) {
dataList.remove(0);
return null;
}
TreeNode root = new TreeNode(Integer.valueOf(dataList.get(0)));
dataList.remove(0);
root.left = rdeserialize(dataList);
root.right = rdeserialize(dataList);
return root;
}
- 整个序列和反序列逻辑上是对称的。