LintCode——7. 二叉树的序列化和反序列化

描述:

设计一个算法，并编写代码来序列化和反序列化二叉树。将树写入一个文件被称为“序列化”，读取文件后重建同样的二叉树被称为“反序列化”。

如何反序列化或序列化二叉树是没有限制的，你只需要确保可以将二叉树序列化为一个字符串，并且可以将字符串反序列化为原来的树结构。

样例:

给出一个测试数据样例， 二叉树{3,9,20,#,#,15,7}，表示如下的树结构：
3
/ \
9 20
/ \
15 7
我们的数据是进行BFS遍历得到的。当你测试结果wrong answer时，你可以作为输入调试你的代码。
你可以采用其他的方法进行序列化和反序列化。

可以参考二叉树的层次遍历: 二叉树的遍历——递归、非递归及层序遍历

实现代码如下：

package lintcode;

import ds.TreeNode;

import java.util.ArrayDeque;
import java.util.Deque;

/**
 * Class Name  : SerializeBT_7
 * Describe    : empty
 * Create User : yoo
 * Create Date : 2018/1/27
 * Create Time : 19:05
 **/

public class SerializeBT_7 {

    public static String serialize(TreeNode root) {
        StringBuilder sb = new StringBuilder();
        if (root == null)
            return null;
        Deque<TreeNode> nodeDeque = new ArrayDeque<TreeNode>();
        nodeDeque.add(root);
        sb.append(root.val);
        while (!nodeDeque.isEmpty()) {
            TreeNode p = nodeDeque.pop();
            if (p.left != null) {
                nodeDeque.add(p.left);
                sb.append("," + p.left.val);
            } else {
                sb.append(",#");
            }
            if (p.right != null) {
                nodeDeque.add(p.right);
                sb.append("," + p.right.val);
            } else {
                sb.append(",#");
            }
        }

        String results = sb.toString();
        while (results.endsWith(",#,#"))
            results = results.substring(0, results.length() - 4);

        return "{" + results + "}";
    }


    public static TreeNode deserialize(String data) {
        if (data == null || data.trim().length() < 2)
            return null;
        data = data.substring(1, data.length() - 1);

        String[] datas = data.split(",");
        Deque<TreeNode> nodeDeque = new ArrayDeque<TreeNode>();
        TreeNode root = new TreeNode(Integer.valueOf(datas[0]));
        nodeDeque.add(root);
        int i = 1;
        while (i < datas.length) {
            TreeNode node = nodeDeque.pop();
            if (!datas[i].equals("#")) {
                node.left = new TreeNode(Integer.valueOf(datas[i]));
                nodeDeque.add(node.left);
            } else {
                node.left = null;
            }
            i++;
            if (!(i < datas.length))
                break;
            if (!datas[i].equals("#")) {
                node.right = new TreeNode(Integer.valueOf(datas[i]));
                nodeDeque.add(node.right);
            } else {
                node.right = null;
            }
            i++;
        }
        return root;
    }
}