Leetcode 297-二叉树的序列化与反序列化

题解

题解转载自xiao_ben_zhu

BFS 解法

序列化——很典型的 BFS

维护一个队列，初始让根节点入列，考察出列节点：
如果出列的节点是 null，将符号 ‘X’ 推入 res 数组。
如果出列的节点是数值，将节点值推入数组 res，并将它的左右子节点入列。
子节点 null 也要入列，它对应 “X”，要被记录，只是它没有子节点可入列。
入列、出列…直到队列为空，就遍历完所有节点，res构建完毕，转成字符串就好。

反序列化——也是BFS

下图是BFS得到的序列化字符串，它是一层层的。除了第一个是根节点的值，其他节点值都是成对的，对应左右子节点。

依然先转成list数组，用一个指针 cursor 从第二项开始扫描。
起初，用list[0]构建根节点，并让根节点入队。
节点出队，此时 cursor 指向它的左子节点值，cursor+1 指向它的右子节点值。
如果子节点值是数值，则创建节点，并认出列的父亲，同时自己也是父亲，入列。
如果子节点值为 ‘X’，什么都不用做，因为出列的父亲的 left 和 right 本来就是 null
可见，所有的真实节点都会在队列里走一遍，出列就带出儿子入列

public class Codec {

    // Encodes a tree to a single string.
    //层次遍历
    public String serialize(TreeNode root) {
        if(root == null){
            return "";
        }
        StringBuilder res = new StringBuilder();
        res.append("[");
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            TreeNode node = queue.poll();
            if(node != null){
                res.append("" + node.val);
                queue.offer(node.left);
                queue.offer(node.right);
            }else{
                res.append("null");
            }
            res.append(",");
        }
    return res.toString();
        
    }

    // Decodes your encoded data to tree.

    public TreeNode deserialize(String data) {
        if(data == ""){
            return null;
        }
        //去掉'[]'，去掉','
        String[] dataList = data.substring(1, data.length() - 1).split(",");
        //给根节点赋值
        TreeNode root = new TreeNode(Integer.parseInt(dataList[0]));
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        //从根节点的下一个节点开始遍历
        int i=1;
        while(!queue.isEmpty()&&i<dataList.length){
            TreeNode node = queue.poll();
            //字符串的判断要用equals而不是==
            if(!dataList[i].equals("null")){
                //给node.left赋值后入队,下一次遍历从该层最左边node开始赋值
                node.left= new TreeNode(Integer.parseInt(dataList[i++]));
                queue.offer(node.left);
            }else{
                //也可以什么都不做，只i++
                node.left=null;
                i++;
            }
            if(!dataList[i].equals("null")){
                //给node的右节点赋值后入队
                node.right= new TreeNode(Integer.parseInt(dataList[i++]));
                queue.offer(node.right);
            }else{
                node.right=null;
                i++;
            }
        }
        return root;
    }

   
}