forked from Google-DSC-TMSL/ProjectAlgorithms
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Serialize and Deserialize BST
104 lines (93 loc) · 2.73 KB
/
Serialize and Deserialize BST
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
Serialization is 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 search tree. There is no restriction on how your serialization/deserialization algorithm should work. You need to ensure that a binary search tree can be serialized to a string, and this string can be deserialized to the original tree structure.
The encoded string should be as compact as possible.
Input: root = [2,1,3]
Output: [2,1,3]
/**
* 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) {
if(root == NULL)
{
return("");
}
string s = "";
queue<TreeNode*>q;
q.push(root);
while(q.size()>0)
{
TreeNode* m = q.front();
q.pop();
if(m == NULL)
{
s.append("#,");
}
else
{
s.append(to_string(m->val)+",");
}
if(m!= NULL)
{
q.push(m->left);
q.push(m->right);
}
}
cout<<s<<endl;
return(s);
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data) {
if(data.size()== 0)
{
return(NULL);
}
stringstream s(data);
string str;
queue<TreeNode*>q;
getline(s,str,',');
TreeNode* m = new TreeNode(stoi(str));
q.push(m);
while(q.size()>0)
{
TreeNode* n =q.front();
q.pop();
getline(s,str,',');
if(str == "#")
{
n->left = NULL;
}
else
{
TreeNode* n1 = new TreeNode(stoi(str));
n->left = n1;
q.push(n1);
}
getline(s,str,',');
if(str == "#")
{
n->right = NULL;
}
else
{
TreeNode* n2 = new TreeNode(stoi(str));
n->right = n2;
q.push(n2);
}
}
return(m);
}
};
// Codec* ser = new Codec();
// Codec* deser = new Codec();
// string tree = ser->serialize(root);
// TreeNode* ans = deser->deserialize(tree);
// return ans;