2020-11-8 二叉树的序列化

题目描述

二叉树被记录为文件的过程叫作二叉树的序列化，通过文件内容重建原来二叉树的过程叫作二叉树的反序列化，给定一颗二叉树，请将该二叉树先序序列化和层序序列化。

假设序列化的结果字符串为 str，初始时 str = ""，遍历二叉树时，遇到 null 节点，则在 str 的末尾加上 "#!"，否则加上"当前的节点值!"。

输入描述:

第一行输入两个整数 n 和 root，n 表示二叉树的总节点个数，root 表示二叉树的根节点。
以下 n 行每行三个整数 fa，lch，rch，表示 fa 的左儿子为 lch，右儿子为 rch。(如果 lch 为 0 则表示 fa 没有左儿子，rch同理)

输出描述:

输出两行分别表示该二叉树的先序序列化和层序序列化

示例1

输入

复制

2 1
1 2 0
2 0 0

输出

复制

1!2!#!#!#!
1!2!#!#!#!

备注:

1≤n≤1000000
1≤fa,lch,rch,root≤n

#include <bits/stdc++.h>
using namespace std;


struct TreeNode {
    int val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(int value):val(value), left(nullptr), right(nullptr) {}
};

void createTree(TreeNode* root, int& cnt){
    if(cnt == 0) return;
    int p, l, r;
    cin >> p >> l >> r;
     
    if(l){
        TreeNode* lch = new TreeNode(l);
        root->left = lch;
        createTree(root->left, --cnt);
    }
    if(r){
        TreeNode* rch = new TreeNode(r);
        root->right = rch;
        createTree(root->right, --cnt);
    }
    return;
}
//递归版
// string preOrderTravel(TreeNode* root) {
//     if(!root)
//         return "#!";
//     string res = to_string(root->val) + "!";
//     res += preOrderTravel(root->left);
//     res += preOrderTravel(root->right);
//     return res;
// }
string preOrderTravel(TreeNode* root) {
    if(root == nullptr) {
        return "#!";
    }
    TreeNode* cur = root;
    string res = "";
    stack<TreeNode*> treeStack;
    treeStack.push(cur);
    while(!treeStack.empty()) {
        cur = treeStack.top();
        treeStack.pop();
        if(cur == nullptr) {
            res += "#!";
        } else {
            res += to_string(cur->val) + "!";
            treeStack.push(cur->right);
            treeStack.push(cur->left);
        }
    }
    return res;
}


string levelOrderTravel(TreeNode* root) {
    if(!root) 
        return "#!";
    queue<TreeNode*> treeQueue;
    TreeNode* cur = root;
    treeQueue.push(cur);
    string res = "";
    while(!treeQueue.empty()) {
        cur = treeQueue.front();
        treeQueue.pop();
        if(cur == nullptr) {
            res += "#!";
        } else {
            res += to_string(cur->val) + "!";
            treeQueue.push(cur->left);
            treeQueue.push(cur->right);
        }
    }
    return res;
}
int main() {
    int n, r;
    cin >> n >> r;
    TreeNode* root = new TreeNode(r);
    createTree(root, n);
    string res = preOrderTravel(root);
    cout << res << endl;
    string ret = levelOrderTravel(root);
    cout << ret << endl;
    
    return 0;
}