请实现两个函数,分别用来序列化和反序列化二叉树。
示例:
你可以将以下二叉树:
1
/ \
2 3
/ \
4 5
序列化为 “[1,2,3,null,null,4,5]”
注意:本题与主站 297 题相同:https://leetcode-cn.com/problems/serialize-and-deserialize-binary-tree/
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/xu-lie-hua-er-cha-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
BFS遍历,和BFS的构建。主要的难点在于如何根据自己BFS得出的string逆向的构建出原来的二叉树。思路是,与正向的BFS相似,利用队列,向队列中插入构造好的头节点,出队列的时候,构造好出队列节点的左孩子和右孩子,并将他们加入队列即可。此时的问题即,如何找到出队节点的左右子节点,这时,因为在构造string的过程中,记录下来的所有节点一定是某一个节点的子节点(除根节点之外),同时,除开null节点之外所有的节点一定有两个子节点,因此,在string中的节点的顺序,也是严格的根据bfs顺序来的。设置起始的位置为index = 1,则index++则为右节点,在index++则移动到队列中下一个元素的左节点的位置。
/**
* 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)
{
queue<TreeNode*> q;
if(root==nullptr)return "";
string ret;
q.push(root);
while(!q.empty())
{
TreeNode* temp = q.front();
q.pop();
if(temp!=nullptr)
{
ret+=to_string(temp->val);
q.push(temp->left);
q.push(temp->right);
}
else
{
ret+="null";
}
if(!q.empty())
{
ret+=",";
}
}
return ret;
}
// Decodes your encoded data to tree.
TreeNode* deserialize(string data)
{
if(data=="")return nullptr;
vector<string> strs;
string temp;
for(auto ch:data)
{
if(ch==',')
{
strs.push_back(temp);
temp.clear();
continue;
}
else
{
temp+=ch;
}
}
strs.push_back(temp);
queue<TreeNode*> q;
TreeNode* root = new TreeNode(stoi(strs[0]));
int index = 1;
q.push(root);
while(!q.empty())
{
TreeNode* temp = q.front();
q.pop();
if(strs[index]!="null")
{
temp->left = new TreeNode(stoi(strs[index]));
q.push(temp->left);
}
index++;
if(strs[index]!="null")
{
temp->right = new TreeNode(stoi(strs[index]));
q.push(temp->right);
}
index++;
}
return root;
}
};
// Your Codec object will be instantiated and called as such:
// Codec codec;
// codec.deserialize(codec.serialize(root));