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

297. 二叉树的序列化与反序列化

序列化时，若当前节点为空，则加入 # 号并返回。否则，加入当前节点值，再递归左右子树。

反序列化时，先将字符串转换为列表，再确定根节点，即列表首个元素，再递归生成左右子树。

public class Solution {
    
    StringBuilder sb = new StringBuilder();
    
    String Serialize(TreeNode root) {
        serialize(root);
        return sb.toString();
    }
    
    /* 辅助函数，将二叉树存入 StringBuilder */
    void serialize(TreeNode root) {
        if (root == null) {
            sb.append("#,");
            return;
        }
        sb.append(root.val + ",");
        serialize(root.left);
        serialize(root.right);
    }
    
    TreeNode Deserialize(String data) {
        // 将字符串转化成 String 类型列表
        LinkedList<String> nodes = new LinkedList();
        for (String s : data.split(","))
            nodes.addLast(s);
        return deserialize(nodes);
    }
    
    /* 辅助函数，通过 nodes 列表构造二叉树 */
    TreeNode deserialize(LinkedList<String> nodes) {
        if (nodes.isEmpty()) return null;
        
        // 列表最左侧就是根节点
        String first = nodes.removeFirst();
        if (first.equals("#")) return null;
        TreeNode root = new TreeNode(Integer.parseInt(first));
        root.left = deserialize(nodes);
        root.right = deserialize(nodes);
        
        return root;
    }
}