题目描述
请实现两个函数,分别用来序列化和反序列化二叉树。
示例:
你可以将以下二叉树:
1
/ \
2 3
/ \
4 5序列化为 "[1,2,3,null,null,4,5]"
参考解题思路: 序列化用dfs层序遍历,反序列化就需要在序列化时携带二叉树的完整信息
序列化 Serialize :借助队列,对二叉树做层序遍历,并将越过叶节点的 null 也打印出来。
算法流程:
- 特例处理: 若 root 为空,则直接返回空列表 "[]" ;
- 初始化: 队列 queue (包含根节点 root );序列化列表 res ;
- 层序遍历: 当 queue 为空时跳出;
- 节点出队,记为 node ;
- 若 node 不为空:① 打印字符串 node.val ,② 将左、右子节点加入 queue ;
- 否则(若 node 为空):打印字符串 "null" ;
- 返回值: 拼接列表,用 ',' 隔开,首尾添加中括号;
复杂度分析:
时间复杂度 O(N): N为二叉树的节点数,层序遍历需要访问所有节点,最差情况下需要访问 N+1 个 null ,总体复杂度为 O(2N + 1) = O(N) 。
空间复杂度 O(N): 最差情况下,队列 queue 同时存储(N + 1)/2个节点(或 N+1 个 null ),使用 O(N) ;列表 res 使用 O(N) 。
反序列化 Deserialize :基于本文开始推出的 node , node.left , node.right 在序列化列表中的位置关系,可实现反序列化。利用队列按层构建二叉树,借助一个指针 i 指向节点 node 的左、右子节点,每构建一个 node 的左、右子节点,指针 i 就向右移动 11 位。
算法流程:
- 特例处理: 若 data 为空,直接返回 null ;
- 初始化: 序列化列表 vals (先去掉首尾中括号,再用逗号隔开),指针 i = 1 ,根节点 root (值为 vals[0] ),队列 queue(包含 root );
- 按层构建: 当 queue 为空时跳出;
- 节点出队,记为 node ;
- 构建 node 的左子节点:node.left 的值为 vals[i] ,并将 node.left 入队;
- 执行 i += 1 ;
- 构建 node 的右子节点:node.left 的值为 vals[i] ,并将 node.left 入队;
- 执行 i += 1 ;
- 返回值: 返回根节点 root 即可;
复杂度分析:
时间复杂度 O(N): N 为二叉树的节点数,按层构建二叉树需要遍历整个vals ,其长度最大为2N+1 。
空间复杂度 O(N) : 最差情况下,队列 queue 同时存储 (N + 1)/2个节点,因此使用O(N) 额外空间。
// Encodes a tree to a single string.
public String serialize(TreeNode root) {
// 边界值处理
if (root == null) {
return "[]";
}
StringBuilder res = new StringBuilder("[");
Deque<TreeNode> deque = new LinkedList<>();
deque.offer(root);
while (!deque.isEmpty()) {
TreeNode node = deque.pop(); // 队列弹出节点
if (node != null) {
res.append(node.val + ","); // 不为空则追加节点值进字符串并加上,
// 添加节点的左右子节点进队列
deque.offer(node.left);
deque.offer(node.right);
} else {
// 为空则添加null + ,
res.append("null,");
}
}
res.deleteCharAt(res.length()-1); // 删除最后多余的,
return res.append("]").toString(); // 添加饭括弧进字符串,转为string
}
// Decodes your encoded data to tree.
public TreeNode deserialize(String data) {
// 边界值处理--空字符串返回空二叉树
if (data.equals("[]")) {
return null;
}
// 截取[]以内的字符串以,为分隔符将其转换为String数组
String[] vals = data.substring(1,data.length()-1).split(",");
// 创建根节点 -- 为string数组的首位值转int
TreeNode root = new TreeNode(Integer.parseInt(vals[0]));
Deque<TreeNode> deque = new LinkedList(); // 创建节点队列
deque.offer(root); // 放入根节点
int i = 1; // 遍历数组
while (!deque.isEmpty()){
TreeNode node = deque.pop(); // 队列弹出节点
// 数组下标不为空
if (!vals[i].equals("null")) {
// 添加二叉树左节点
node.left = new TreeNode(Integer.parseInt(vals[i]));
// 队列新加节点
deque.offer(node.left);
}
i++; // 数组下标加一
// 数组下标不为空
if (!vals[i].equals("null")) {
// 添加二叉树右节点
node.right = new TreeNode(Integer.parseInt(vals[i]));
// 队列新加节点
deque.offer(node.right);
}
i++; // 数组下标加一
}
return root;
}
作者:jyd
链接:https://leetcode-cn.com/problems/xu-lie-hua-er-cha-shu-lcof/solution/mian-shi-ti-37-xu-lie-hua-er-cha-shu-ceng-xu-bian-/