题目要求

Serialization is the process of converting a data structure or object into a sequence of bits so that it can be stored in a file or memory buffer, or transmitted across a network connection link to be reconstructed later in the same or another computer environment.Design an algorithm to serialize and deserialize a binary search tree. There is no restriction on how your serialization/deserialization algorithm should work. You just need to ensure that a binary search tree can be serialized to a string and this string can be deserialized to the original tree structure.The encoded string should be as compact as possible.Note: Do not use class member/global/static variables to store states. Your serialize and deserialize algorithms should be stateless.

将二叉搜索树序列化和反序列化,序列化是指将树用字符串的形式表示,反序列化是指将字符串形式的树还原成原来的样子。

思路和代码

对于树的序列化,可以直接联想到对树的遍历。树的遍历包括前序遍历,中序遍历,后序遍历和水平遍历,并且可知前序遍历和中序遍历,或中序遍历和后序遍历可以构成一棵唯一的树。除此以外,因为这是一棵二叉搜索树,可知该树的中序遍历就是所有元素的从小到大的排列。

举个例子,假如一棵树的结构如下:

  3 / \2   4 \  1

该树的前序遍历结果为3,2,1,4,中序遍历为1,2,3,4。再仔细分析前序遍历的结果,结合二叉搜索树可知,比中间节点小的值一定位于左子树,反之一定位于右子树,即可以对前序遍历进行分割3,|2,1,|4。也就是说,我们可以只利用前序遍历,就可以区分出二叉搜索树的左子树和右子树。
代码如下:

    public String serialize(TreeNode root) {        StringBuilder sb = new StringBuilder();        preorder(root, sb);        return sb.toString();    }    public void preorder(TreeNode root, StringBuilder result) {        if(root != null) {            result.append(root.val);            result.append(":");            preorder(root.left, result);            preorder(root.right, result);        }    }    // Decodes your encoded data to tree.    public TreeNode deserialize(String data) {        if(data==null || data.isEmpty()) return null;        String[] preorder = data.split(":");        String[] inorder = Arrays.copyOf(preorder, preorder.length);        Arrays.sort(inorder, new Comparator<String>(){            @Override            public int compare(String o1, String o2) {                Integer i1 = Integer.valueOf(o1);                Integer i2 = Integer.valueOf(o2);                return i1.compareTo(i2);            }                    });                return build(inorder, preorder, 0, 0, inorder.length);    }        public TreeNode build(String[] inorder, String[] preorder, int inorderStart, int preorderStart, int length) {        if(length <= 0) return null;        TreeNode root = new TreeNode(Integer.valueOf(preorder[preorderStart]));        for(int i = inorderStart ; i < inorderStart+length ; i++) {            if(inorder[i].equals(preorder[preorderStart])) {                root.left = build(inorder, preorder, inorderStart, preorderStart+1, i-inorderStart);                root.right = build(inorder, preorder, i+1, preorderStart+i-inorderStart+1, inorderStart+length-i-1);                break;            }        }                return root;    }

这里的代码是直接使用排序生成了二叉搜索树的中序遍历的结果,并利用先序遍历和中序遍历构造了一棵二叉搜索树。假如二叉搜索树的节点较多,该算法将会占用大量的额外空间。可以只用先序遍历作为构造树的输入,代码如下:

    public TreeNode deserialize(String data) {        if (data==null) return null;        String[] strs = data.split(":");        Queue<Integer> q = new LinkedList<>();        for (String e : strs) {            q.offer(Integer.parseInt(e));        }        return getNode(q);    }        private TreeNode getNode(Queue<Integer> q) {        if (q.isEmpty()) return null;        TreeNode root = new TreeNode(q.poll());//root (5)        Queue<Integer> samllerQueue = new LinkedList<>();        while (!q.isEmpty() && q.peek() < root.val) {            samllerQueue.offer(q.poll());        }        root.left = getNode(samllerQueue);        root.right = getNode(q);        return root;    }