手撕LeetCode 二叉树的序列化和反序列化
扣第 297 题「二叉树的序列化与反序列化」
就是给你输入一棵二叉树的根节点 root,要求你实现如下一个类:
public class Codec { // 把一棵二叉树序列化成字符串 public String serialize(TreeNode root) { } // 把字符串反序列化成二叉树 public TreeNode deserialize(String data) { } }
我们可以用 serialize 方法将二叉树序列化成字符串,用 deserialize 方法将序列化的字符串反序列化成二叉树,至于以什么格式序列化和反序列化,这个完全由你决定。
比如说输入如下这样一棵二叉树:
serialize 方法也许会把它序列化成字符串 2,1,#,6,3,#,#,其中 # 表示 null 指针,那么把这个字符串再输入 deserialize 方法,依然可以还原出这棵二叉树。也就是说,这两个方法会成对儿使用,你只要保证他俩能够自洽就行了。
前序遍历的实现方式
public class Codec { String SEQ=","; String NULL="#"; // 把一棵二叉树序列化成字符串 public String serialize(TreeNode root) { StringBuilder sb=StringBuilder(); serialize(root,sb); return sb.toString; } //自定义辅助函数,将 public void serialize(TreeNode root,StringBuilder sb){ if(root==null) { sb.append(NULL).append(SEQ); return;} sb.append(root.val).append(SEQ); serialize(root.left,sb); serialize(root.right,sb); } // 把字符串反序列化成二叉树 public TreeNode deserialize(String data) { LinkedList<String> nodes=new LinkedList<>(); for(String s:data.split(SEQ){ nodes.add(s); } return deserialize(nodes); } //利用前序遍历使列表转变为树 public TreeNode deserialize(LinkedList<String> nodes){ if(nodes.isEmpty()) return null; String first=nodes.poll(); if(first.equals(NULL)) return null; TreeNode root=new TreeNode(Integer.parseInt(first)); root.left=deserialize(nodes); root.right=deserialize(nodes); return root; } }
上一篇:
IDEA上Java项目控制台中文乱码