P1305 新二叉树——构建二叉树，前序遍历

题目描述

输入一串二叉树，输出其前序遍历。

输入格式

第一行为二叉树的节点数 $n$。($1\le n\le 26$)

后面 $n$ 行，每一个字母为节点，后两个字母分别为其左右儿子。特别地，数据保证第一行读入的节点必为根节点。

空节点用 * 表示

输出格式

二叉树的前序遍历。

输入输出样例 #1

输入 #1

6
abc
bdi
cj*
d**
i**
j**

输出 #1

abdicj

题解

#include <bits/stdc++.h>
using namespace std;
vector<char> res;
struct TreeNode {
    char val;
    TreeNode* left;
    TreeNode* right;
    TreeNode(char x):val(x),left(nullptr),right(nullptr) {}
};
void dfs(TreeNode* root) {
    if(root==nullptr) return;
    res.push_back(root->val);
    dfs(root->left);
    dfs(root->right);
}
int main() {
    int n;
    cin>>n;
    unordered_map<char,TreeNode*> tree;
    unordered_set<char> myset;
    TreeNode* root=nullptr;
    for (int i=0;i<n;i++) {
        char a,b,c;
        cin>>a>>b>>c;
        if(!myset.count(a)) {
            tree[a]=new TreeNode(a);
            if(i==0) root=tree[a];
        }
        if(!myset.count(b)&&b!='*') tree[b]=new TreeNode(b);
        if(!myset.count(c)&&c!='*') tree[c]=new TreeNode(c);
        if(b!='*') tree[a]->left=tree[b];
        if(c!='*') tree[a]->right=tree[c];
        myset.insert(a);myset.insert(b);myset.insert(c);
    }
    dfs(root);
    for (char ch:res) {
        cout<<ch;
    }
}