/
LC_00297_SerializeAndDeserializeBinaryTree.cpp
110 lines (98 loc) · 3.31 KB
/
LC_00297_SerializeAndDeserializeBinaryTree.cpp
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
/*
Serialization is the process of converting a data structure or object into a sequence of bits so that it can be stored in a file or memory buffer, or transmitted across a network connection link to be reconstructed later in the same or another computer environment.
Design an algorithm to serialize and deserialize a binary tree. There is no restriction on how your serialization/deserialization algorithm should work. You just need to ensure that a binary tree can be serialized to a string and this string can be deserialized to the original tree structure.
For example, you may serialize the following tree
1
/ \
2 3
/ \
4 5
as "[1,2,3,null,null,4,5]", just the same as how LeetCode OJ serializes a binary tree. You do not necessarily need to follow this format, so please be creative and come up with different approaches yourself.
Note: Do not use class member/global/static variables to store states. Your serialize and deserialize algorithms should be stateless.
Credits:
Special thanks to @Louis1992 for adding this problem and creating all test cases.
Hide Tags Tree Design
*/
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Codec {
public:
// Encodes a tree to a single string.
string serialize(TreeNode* root) {
ostringstream out;
stack<TreeNode*> stk;
stk.push(root);
while (stk.size()) {
TreeNode* node = stk.top();
stk.pop();
if (node) {
out << node->val << ' ';
stk.push(node->right);
stk.push(node->left);
} else {
out << "# ";
}
}
return out.str();
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data) {
istringstream in(data);
return deserializeHelper(in);
}
private:
TreeNode* deserializeHelper(istringstream& in) {
string val;
in >> val;
if (val == "#")
return NULL;
TreeNode* root = new TreeNode(stoi(val));
root->left = deserializeHelper(in);
root->right = deserializeHelper(in);
return root;
}
};
class Codec {
public:
// Encodes a tree to a single string.
string serialize(TreeNode* root) {
ostringstream out;
serializeHelper(root, out);
return out.str();
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data) {
istringstream in(data);
return deserializeHelper(in);
}
private:
void serializeHelper(TreeNode* root, ostringstream& out) {
if (root) {
out << root->val << ' ';
serializeHelper(root->left, out);
serializeHelper(root->right, out);
} else {
out << "# ";
}
}
TreeNode* deserializeHelper(istringstream& in) {
string val;
in >> val;
if (val == "#")
return NULL;
TreeNode* root = new TreeNode(stoi(val));
root->left = deserializeHelper(in);
root->right = deserializeHelper(in);
return root;
}
};
// Your Codec object will be instantiated and called as such:
// Codec codec;
// codec.deserialize(codec.serialize(root));