剑指 Offer 37. 序列化二叉树

题目

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

你需要设计一个算法来实现二叉树的序列化与反序列化。这里不限定你的序列 / 反序列化算法执行逻辑,你只需要保证一个二叉树可以被序列化为一个字符串并且将这个字符串反序列化为原始的树结构。

提示:输入输出格式与 LeetCode 目前使用的方式一致,详情请参阅 LeetCode 序列化二叉树的格式。你并非必须采取这种方式,你也可以采用其他的方法解决这个问题。

示例:

img
1
2
输入:root = [1,2,3,null,null,4,5]
输出:[1,2,3,null,null,4,5]

思路

层序遍历,然后在依次构建

实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
private static class Codec {

// Encodes a tree to a single string.
public String serialize(TreeNode root) {
if (root == null) {
return "";
}
// 定义一个队列保存每层节点
LinkedList<TreeNode> deque = new LinkedList<>();
// 用一个list接收结果,最后转为string
StringBuilder result = new StringBuilder();
deque.add(root);
boolean append = true;
// 循环遍历
while (!deque.isEmpty()) {
TreeNode cur = deque.pop();
if (cur == null) {
result.append("null,");
append = false;
continue;
}
result.append(cur.val).append(",");
deque.add(cur.left);
deque.add(cur.right);
}


return result.substring(0, result.length() - 1);
}

// Decodes your encoded data to tree.
public TreeNode deserialize(String data) {
if (data.length() == 0) {
return null;
}
String[] rootArr = data.split(",");
int size = rootArr.length;
for (int i = size - 1; i >= 0; i--) {
if (!"null".equals(rootArr[i])) {
size--;
} else {
break;
}
}

int index = 0;
TreeNode root = new TreeNode(Integer.parseInt(rootArr[0]));
index++;
LinkedList<TreeNode> deque = new LinkedList<>();
deque.add(root);

while (index < size) {
TreeNode cur = deque.pop();
if (!"null".equals(rootArr[index])) {
TreeNode rootLeft = new TreeNode(Integer.parseInt(rootArr[index]));
cur.left = rootLeft;
deque.add(rootLeft);
}
index++;
if (!"null".equals(rootArr[index])) {
TreeNode rootRight = new TreeNode(Integer.parseInt(rootArr[index]));
cur.right = rootRight;
deque.add(rootRight);
}
index++;
}

return root;
}
}