二叉树的序列化和反序列化

二叉树的序列化和反序列化

题目:二叉树的序列化

《程序员代码面试指南》第34题 P107 难度:士★☆☆☆

二叉树的序列化就是二叉树被记录成文件的过程,二叉树的反序列化就是通过文件内容重建原来二叉树的过程。

序列化过程:首先假设序列化的结果字符串为str,初始时str=""。如果遇到null节点,就在str的末尾加上"#!","#"表示这个节点为,节点不存在,"!"表示一个值的结束;如果遇到不为空的节点,假设节点值为2,就在str的末尾加上"2!"。

本书介绍了两种序列化和反序列化的实现。

首先是先序遍历序列化过程如下:

public String serialByPre(Node head) {
    if (head == null) {
        return "#!";
    }
    String res = head.value + "!";
    res += serialByPre(head.left);
    res += serialByPre(head.right);
    return res;
}

可以看出,该过程很简单,基本就是在先序遍历递归的基础上稍作修改。注意返回值的使用。

然后是反序列化的过程:

public Node reconByPreString(String preStr) {
    String[] values = preStr.split("!");
    Queue<String> queue = new LinkedList<String>();
    for (int i = 0; i != values.length; i++) {
        queue.offer(values[i]);
    }
    return reconPreOrder(queue);
}

public Node reconPreOrder(Queue<String> queue) {
    String value = queue.poll();
    if (value.equals("#")) {
        return null;
    }
    Node head = new Node(Integer.valueOf(value));
    head.left = reconPreOrder(queue);
    head.right = reconPreOrder(queue);
    return head;
}

通过队列+递归来实现。先将所有节点逐一插入队列,然后在逐一取出队列元素的同时,通过判断"#"来选择返回空,或是给当前节点的左孩子或右孩子赋值。

注意,首先需要通过"!"将原来的字符串拆解成字符串数组。同时,递归里层的queue弹出了元素,在外层queue同样有相应的变化。(传值的是引用,内容做了修改,所有为该引用的变量访问的都是同样的内容

其次是层序遍历,序列化和反序列化都用到队列结构。

序列化

public String serialByLevel(Node head) {
    if (head == null) {
        return "#!";
    }
    String res = head.value + "!";
    Queue<Node> queue = new LinkedList<Node>();
    queue.offer(head);
    while (!queue.isEmpty()) {
        head = queue.poll();
        if (head.left != null) {
            res += head.left.value + "!";
            queue.offer(head.left);
        } else {
            res += "#!";
        }
        if (head.right != null) {
            res += head.right.value + "!";
            queue.offer(head.right);
        } else {
            res += "#!";
        }
    }
    return res;
}

反序列化

public Node reconByLevelString(String levelStr) {
    String[] values = levelStr.split("!");
    int index = 0;
    Node head = generateNodeByString(values[index++]);
    Queue<Node> queue = new LinkedList<Node>();
    if (head != null) {
        queue.offer(head);
    }
    Node node = null;
    while (!queue.isEmpty()) {
        node = queue.poll();
        node.left = generateNodeByString(values[index++]);
        node.right = generateNodeByString(values[index++]);
        if (node.left != null) {
            queue.offer(node.left);
        }
        if (node.right != null) {
            queue.offer(node.right);
        }
    }
    return head;
}

public Node generateNodeByString(String val) {
    if (val.equals("#")) {
        return null;
    }
    return new Node(Integer.valueOf(val));
}

对于两种遍历的反序列化,其实都是重做相应的遍历,遇到"#"就生成null节点先序遍历就停止生成后续子树的过程,层序遍历就不把null节点放到队列中

算法细节不再赘述,看代码进行理解(⊙o⊙)…

上一篇:Java 泛型类型


下一篇:【LeetCode学习计划】《数据结构入门-C++》第11天 树