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

Serialize and deserilize a binary Tree

WAP 线上面试题 livecoding

Design an algorithm to serialize and deserialize a binary tree, in which the tree can be serialized to a list of Integer and this list can be deserilized to the original tree structure.



并不是leetcode原题，是层序遍历。

一般的是序列化为String，如果是序列化为String，就需要在 node之间插入’!’ 表示结束。

然后’#’表示null。

但本题是list，思想类似，不过处理null有点麻烦.

live coding的时候紧张lol， 面试官都给tips，序列化和反序列化都是层序化，反序列化注意root要先存到queue，按顺序赋给left 和right即可。

代码如下：

package worksExam;
import java.util.*;

class TreeNode{
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val){
this.val=val;
}
}

public class Main {

public List<Integer> serialize(TreeNode root){
List<Integer> list=new ArrayList<Integer>();
if(root==null)
return list;
LinkedList<TreeNode> queue=new LinkedList<TreeNode>();
queue.add(root);
while(queue.isEmpty()==false){
TreeNode treeNode=queue.peek();
queue.remove();
/** -1 表示该节点是null！
* 如果是表示null，加入null到list，
* 下面就不要执行了，直接continue到下一次remove。
* 否则就会进入死循环，所以后面不会有的就continue！
* */
if(treeNode.val!=-1){
list.add(treeNode.val);
}else{
list.add(null);
continue;
}

if(treeNode.left!=null){
queue.add(treeNode.left);
}
else{
queue.add(new TreeNode(-1));
}
if(treeNode.right!=null){
queue.add(treeNode.right);

}
else{
queue.add(new TreeNode(-1));
}

}
return list;
}

public TreeNode deserialize(List<Integer> data){
if(data.size()<1||data==null)
return null;
//root
LinkedList<TreeNode> queue=new LinkedList<TreeNode>();
TreeNode root=new TreeNode(data.get(0));
int index=0;
queue.add(root);
TreeNode node=null;
while(!queue.isEmpty()){
node=queue.remove();
if(data.get(++index)!=null)
node.left=new TreeNode(data.get(index));
if(data.get(++index)!=null)
node.right=new TreeNode(data.get(index));
if(node.left!=null){
queue.add(node.left);
}
if(node.right!=null){
queue.add(node.right);
}

}
return root;
}

}