1 二叉树遍历

#include
#include
#include

using namespace std;

struct TreeNode
{
    char data;
    TreeNode* leftChild;
    TreeNode* rightChild;
};

TreeNode* createTreeNode(const char* str)
{
    stack s;
    TreeNode* root = nullptr;
    TreeNode* currentNode = nullptr;
    bool isleftChild = false;

    for(int i = 0; str[i] != '\0'; i++)
    {
        char ch = str[i];

        switch(ch)
        {
            case '{':
                s.push(currentNode);
                isleftChild = true;
                break;
            case ',':
                isleftChild = false;
                break;
            case '}':
                s.pop();
                break;
            default:
                currentNode = new TreeNode();
                currentNode->data = ch;

                if(root == nullptr)
                {
                    root = currentNode;
                }
                else
                {
                    if(isleftChild)
                    {
                        s.top()->leftChild = currentNode;
                    }
                    else
                    {
                        s.top()->rightChild = currentNode;
                    }
                }
                break;
        }
    }

    return root;
}

void inOrderTraversal(TreeNode* root)
{
    if(root == nullptr)
    {
        return;
    }
    else
    {
        inOrderTraversal(root->leftChild);
        cout << root->data << " ";
        inOrderTraversal(root->rightChild);
    }
}

int main()
{
    string inputStr;
    getline(cin, inputStr);

    TreeNode* root = createTreeNode(inputStr.c_str());

    inOrderTraversal(root);

    cout << endl;

    return 0;
}
 

你可能感兴趣的:(代码,c++,算法,开发语言)