-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathLC297.cpp
123 lines (99 loc) · 3.13 KB
/
LC297.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
111
112
113
114
115
116
117
118
119
120
121
122
123
/*
297. Serialize and Deserialize Binary Tree
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.
Example:
You may serialize the following tree:
1
/ \
2 3
/ \
4 5
as "[1,2,3,null,null,4,5]"
Clarification: The above format is the same as how LeetCode serializes a binary tree.
You do not necessarily need to follow this format, so please be creative and come up with different approaches yourself.
/**
* 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)
{
std::queue<TreeNode*> Q;
Q.push(root);
std::string serialized;
while (!Q.empty())
{
TreeNode* top = Q.front();
if (top == NULL)
{
serialized.append("NULL ");
Q.pop();
}
else
{
serialized.append(std::to_string(top->val) + " ");
Q.pop();
Q.push(top->left);
Q.push(top->right);
}
}
serialized.pop_back();
return serialized;
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data)
{
std::vector<std::string> nodes;
std::queue<TreeNode* > Q;
int i = 1;
std::istringstream ss(data);
while(ss)
{
std::string node;
ss >> node;
nodes.push_back(node);
}
if (nodes.front() == "NULL")
return NULL;
TreeNode* node = new TreeNode(std::stoi(nodes.front()));
Q.push(node);
while (!Q.empty() && i < nodes.size())
{
TreeNode* next = Q.front();
Q.pop();
if (nodes[i] == "NULL")
next->left = NULL;
else
{
next->left = new TreeNode(std::stoi(nodes[i]));
Q.push(next->left);
}
i++;
if (nodes[i] == "NULL")
next->right = NULL;
else
{
next->right = new TreeNode(stoi(nodes[i]));
Q.push(next->right);
}
i++;
}
return node;
}
};
// Your Codec object will be instantiated and called as such:
// Codec codec;
// codec.deserialize(codec.serialize(root));