B3642 二叉树的遍历

B3642 二叉树的遍历

题目描述

有一个 $n(n\le 10^6)$ 个结点的二叉树。给出每个结点的两个子结点编号（均不超过 $n$），建立一棵二叉树（根节点的编号为 $1$），如果是叶子结点，则输入 0 0。

建好树这棵二叉树之后，依次求出它的前序、中序、后序列遍历。

输入格式

第一行一个整数 $n$，表示结点数。

之后 $n$ 行，第 $i$ 行两个整数 $l$、$r$，分别表示结点 $i$ 的左右子结点编号。若 $l=0$ 则表示无左子结点，$r=0$ 同理。

输出格式

输出三行，每行 $n$ 个数字，用空格隔开。

第一行是这个二叉树的前序遍历。

第二行是这个二叉树的中序遍历。

第三行是这个二叉树的后序遍历。

输入输出样例 #1

输入 #1

7
2 7
4 0
0 0
0 3
0 0
0 5
6 0

输出 #1

1 2 4 3 7 6 5
4 3 2 1 6 5 7
3 4 2 5 6 7 1

题解

#include <bits/stdc++.h>

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

void preOrder(TreeNode* root, std::vector<int>& result) {
    if(!root) return;
    result.push_back(root->val);
    preOrder(root->left, result);
    preOrder(root->right, result);
}

void inOrder(TreeNode* root, std::vector<int>& result) {
    if(!root) return;
    inOrder(root->left, result);
    result.push_back(root->val);
    inOrder(root->right, result);
}

void postOrder(TreeNode* root, std::vector<int>& result) {
    if(!root) return;
    postOrder(root->left, result);
    postOrder(root->right, result);
    result.push_back(root->val);
}

int main() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);

    int N;
    std::cin >> N;
    std::vector<TreeNode*> T(N + 1);
    for (int i = 1; i <= N; i++) {
        T[i] = new TreeNode(i);
    }
    int left, right;
    for (int i = 1; i <= N; i++) {
        std::cin >> left >> right;
        if(left != 0) T[i]->left = T[left];
        if(right != 0) T[i]->right = T[right];
    }

    TreeNode* root = T[1];
    std::vector<int> preOrder_res, inOrder_res, postOrder_res;
    preOrder(root, preOrder_res);
    inOrder(root, inOrder_res);
    postOrder(root, postOrder_res);

    for(int i = 0; i < preOrder_res.size(); i++) {
        if(i != preOrder_res.size() - 1) std::cout << preOrder_res[i] << " ";
        else std::cout << preOrder_res[i] << std::endl;
    }
    for(int i = 0; i < inOrder_res.size(); i++) {
        if(i != inOrder_res.size() - 1) std::cout << inOrder_res[i] << " ";
        else std::cout << inOrder_res[i] << std::endl;
    }
    for(int i = 0; i < postOrder_res.size(); i++) {
        if(i != postOrder_res.size() - 1) std::cout << postOrder_res[i] << " ";
        else std::cout << postOrder_res[i] << std::endl;
    }

    return 0;
}