第61题——序列化二叉树

265 阅读1分钟

题目:

请实现两个函数,分别用来序列化和反序列化二叉树

二叉树的序列化是指:把一棵二叉树按照某种遍历方式的结果以某种格式保存为字符串,从而使得内存中建立起来的二叉树可以持久保存。序列化可以基于先序、中序、后序、层序的二叉树遍历方式来进行修改,序列化的结果是一个字符串,序列化时通过 某种符号表示空节点(#),以 ! 表示一个结点值的结束(value!)。

二叉树的反序列化是指:根据某种遍历顺序得到的序列化字符串结果str,重构二叉树。

Java

package nowcoder;

public class S61_SerializeAndDeSerialize {
    public int index = -1;
    String Serialize(TreeNode root){
        StringBuffer sb = new StringBuffer();
        if (root == null){
            sb.append("#,");
            return sb.toString();
        }
        sb.append(root.val+",");
        sb.append(Serialize(root.left));
        sb.append(Serialize(root.right));
        return sb.toString();
    }

    TreeNode Deserialize(String str){
        index++;
        int len = str.length();
        if (index >= len)
            return null;
        String[] string = str.split(",");
        TreeNode node = null;
        while (!string[index].equals('#')){
            node = new TreeNode(Integer.valueOf(string[index]));
            node.left = Deserialize(str);
            node.right = Deserialize(str);
        }
        return node;
    }
    public static void main(String[] args){
        S61_SerializeAndDeSerialize s61 = new S61_SerializeAndDeSerialize();
        PrintTreeLayer p = new PrintTreeLayer();
        Integer[] array = {8,6,10,5,7,9,11};
        TreeNode root = p.arrayToTree(array, 0);
        System.out.println(s61.Serialize(root));
    }
}

Python

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None
class SerializeAndDeSerialize:
    def __init__(self):
        self.flag = -1
    def serialize(self, root):
        if not root:
            return '#,'
        return str(root.val)+','+self.serialize(root.left)+self.serialize(root.right)
    def deSerialize(self, str):
        self.flag += 1
        l = str.split(',')
        if self.flag >= len(str):
            return None
        root = None

        if l[self.flag] != '#':
            root = TreeNode(int(l[self.flag]))
            root.left = self.deSerialize(str)
            root.right = self.deSerialize(str)
        return root