请实现两个函数,分别用来序列化和反序列化二叉树。
您需要确保二叉树可以序列化为字符串,并且可以将此字符串反序列化为原始树结构。
数据范围
树中节点数量 [0,1000]。
样例
你可以序列化如下的二叉树
8
/ \
12 2
/ \
6 4
为:"[8, 12, 2, null, null, 6, 4, null, null, null, null]"
所以按照次序的遍历,是可以定位出该节点在树的位置;
故思路就在于:我们使用队列,从根节点开始,层层去构建二叉树的结点。
层序遍历,时间复杂度为O(n)。
// 使用层序遍历的方式序列化二叉树(bfs 版本)
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
string str;
// Encodes a tree to a single string.
string serialize(TreeNode* root) {
if(!root) // 空树的情况
{
str += "null ";
return str;
}
queue<TreeNode*> q;
q.push(root);
while(q.size())
{
auto node = q.front();
q.pop();
if(node == nullptr) str += "null ";
else
{
str += (to_string(node->val) + ' ');
q.push(node->left);
q.push(node->right);
}
}
return str;
}
int getval(string& data, int cur, int next)
{
int val = 0;
if(data[cur] != '-')
for(int i = cur; i < next; ++i) val = val * 10 + data[i] - '0';
else
{
for(int i = cur + 1; i < next; ++i) val = val * 10 + data[i] - '0';
val = -val;
}
return val;
}
// Decodes your encoded data to tree.
// 思路:使用队列,从根节点开始,层层去构建二叉树的结点。
TreeNode* deserialize(string data) {
// 1. 先构建根节点
queue<TreeNode*> q;
auto root = new TreeNode(-1);
int cur = 0, next = 0;
while(next < data.size() && data[next] != ' ') next++; // 此时 next 是第一个空格的位置
if(data[cur] == 'n') return nullptr;
else
{
int val = getval(data, cur, next);
root->val = val;
q.push(root);
}
// 2. 使用队列逐步向下一层扩展(bfs)
cur = next + 1;
next = cur;
while(q.size())
{
auto node = q.front();
q.pop();
if(node == nullptr) continue;
// 解析左节点,解析后链接
TreeNode* left = nullptr;
while(next < data.size() && data[next] != ' ') next++;
if(data[cur] != 'n')
{
int val = getval(data, cur, next);
left = new TreeNode(val);
}
node->left = left;
q.push(left);
cur = next + 1;
next = cur;
// 解析右结点,解析后链接
TreeNode* right = nullptr;
while(next < data.size() && data[next] != ' ') next++;
if(data[cur] != 'n')
{
int val = getval(data, cur, next);
right = new TreeNode(val);
}
node->right = right;
q.push(right);
cur = next + 1;
next = cur;
}
return root;
}
};