-
Notifications
You must be signed in to change notification settings - Fork 368
/
Copy pathSerialize and Deserialize a Binary Tree.java
58 lines (53 loc) · 1.53 KB
/
Serialize and Deserialize a Binary Tree.java
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
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Codec {
private static final String NULL_VALUE = "-";
private static final String DELIMETER = " ";
// Encodes a tree to a single string.
public String serialize(TreeNode root) {
if (root == null) {
return NULL_VALUE;
}
StringBuilder sb = new StringBuilder();
Stack<TreeNode> stack = new Stack<>();
stack.add(root);
while (!stack.isEmpty()) {
TreeNode removed = stack.pop();
if (removed == null) {
sb.append(NULL_VALUE).append(DELIMETER);
continue;
}
sb.append(removed.val).append(DELIMETER);
stack.push(removed.right);
stack.push(removed.left);
}
return sb.toString().trim();
}
// Decodes your encoded data to tree.
public TreeNode deserialize(String data) {
Queue<String> queue = new LinkedList<>(Arrays.asList(data.split("\\s+")));
return helper(queue);
}
private TreeNode helper(Queue<String> queue) {
if (queue.peek().equals(NULL_VALUE)) {
queue.remove();
return null;
}
TreeNode root = new TreeNode(Integer.parseInt(queue.peek()));
queue.remove();
root.left = helper(queue);
root.right = helper(queue);
return root;
}
}
// Your Codec object will be instantiated and called as such:
// Codec ser = new Codec();
// Codec deser = new Codec();
// TreeNode ans = deser.deserialize(ser.serialize(root));