main.cpp

Go to the documentation of this file.

#include <iostream>
#include <vector>
#include <stack>
#include <algorithm>
 
using namespace std;
 
struct TreeNode
{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode() : val(0), left(nullptr), right(nullptr) {}
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
    TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};
 
class Solution
{
public:
    vector<int> inorderTraversal(TreeNode* root)
    {
        vector<int> answer = {};
        stack<TreeNode*> node_stack;
        TreeNode* current = root;
 
        while (current != nullptr || !node_stack.empty())
        {
            while (current != nullptr)
            {
                node_stack.push(current);
                current = current->left;
            }
            current = node_stack.top();
            node_stack.pop();
            answer.push_back(current->val);
            current = current->right;
        }
 
        return answer;
    }
};
 
void delete_tree(TreeNode* root)
{
    if (root == nullptr)
        return;
    delete_tree(root->left);
    delete_tree(root->right);
    delete root;
}
 
int main()
{
    TreeNode* root = new TreeNode(1, new TreeNode(2, new TreeNode(4), new TreeNode(5, new TreeNode(6), new TreeNode(7))), new TreeNode(3, NULL, new TreeNode(8, new TreeNode(9), NULL)));
    Solution sol;
    vector<int> preorder = sol.inorderTraversal(root);
    for (int el: preorder)
        cout << el << ", ";
    cout << endl;
    delete_tree(root);
    return 0;
}