剑指 Offer 37. 序列化二叉树

 

思路：层次遍历

序列化：

　　层次遍历二叉树，注意输出符合题目要求的格式

反序列化：

　　初始化时除去序列化数组的左右中括号，除去逗号，将其转换成一个 String 数组

 

　　这样二叉树中每个节点的索引即为数组的下标

　　按序列化层次遍历的思路，一层一层构建原二叉树。

代码：

序列化与反序列化的时间复杂度均为O（n），空间复杂度均为O（n）

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Codec {
    // Encodes a tree to a single string.
    public String serialize(TreeNode root) {//层次遍历，注意符合题目要求格式
        if (root == null)
            return "[]";
        StringBuilder res = new StringBuilder("[");//首尾加【】
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);
        while (!q.isEmpty()) {
            TreeNode node = q.poll();
            if (node != null) {
                res.append(node.val + ",");
                q.add(node.left);
                q.add(node.right);
            } else
                res.append("null,");//节点为空
        }
        res.deleteCharAt(res.length() - 1);//删除最后一个数字后的逗号
        res.append("]");
        return res.toString();
    }

    // Decodes your encoded data to tree.
    public TreeNode deserialize(String data) {
        if (data.equals("[]"))
            return null;
        String[] vals = data.substring(1, data.length() - 1).split(",");//去掉首尾中括号，去掉逗号，注意substring截取字符串不包括右边界字符
        TreeNode root = new TreeNode(Integer.parseInt(vals[0]));//string转换为int
        Queue<TreeNode> q = new LinkedList<>();
        q.add(root);
        int i = 1;
        while (!q.isEmpty()) {
            TreeNode node = q.poll();
            if (!vals[i].equals("null")) {//添加左子节点
                node.left = new TreeNode(Integer.parseInt(vals[i]));
                q.add(node.left);
            }
            i++;
            if (!vals[i].equals("null")) {//添加右子节点
                node.right = new TreeNode(Integer.parseInt(vals[i]));
                q.add(node.right);
            }
            i++;
        }
        return root;
    }
}