数据结构--后缀表达式转树（c++）

2019独角兽企业重金招聘Python工程师标准>>>

#include <iostream>
#include <vector>
#include <string>
#include <stack>
#include <cctype>  //判断是否是数字、字母或、、

using namespace std;

struct BinaryNode
{
    char element;
    BinaryNode *left;
    BinaryNode *right;
};

typedef BinaryNode * BinaryNodeptr;

void deleteAllNode(BinaryNodeptr ptr)
{
    if (ptr->left!=NULL)
    {
        deleteAllNode(ptr->left);
    }
    else if (ptr->right!=NULL)
    {
        deleteAllNode(ptr->right);
    }
    else
    {
        delete ptr;
        ptr = NULL;
    }
}

class expression_to_tree
{
public:
    expression_to_tree(vector<char> str);
    BinaryNodeptr createNewNode();
    void printTree(BinaryNodeptr node);

    ~expression_to_tree();

    BinaryNodeptr root;

protected:
private:
    vector<char> str;
};

BinaryNodeptr expression_to_tree::createNewNode()
{
    BinaryNodeptr node = new BinaryNode;
    node->left = NULL;
    node->right = NULL;

    return node;
}

expression_to_tree::expression_to_tree(vector<char> str)
{
    stack<BinaryNodeptr> nodeStack;  //保存除运算符之外的字符
    BinaryNodeptr temp_node_ptr;

    char c;

    for (int i = 0; i < str.size(); i++)
    {
        cout << "输入的字符为：" << str[i] <<endl;
        c = str[i];
        temp_node_ptr = createNewNode();
        //temp_node_ptr->element = c;

        if (!(c == '+' || c == '-' || c == '*' || c == '/'))
        {
            cout << "推入栈的字符:" <<c<< endl;
            temp_node_ptr->element = c;
            nodeStack.push(temp_node_ptr);
        }
        else
        {
            cout << "推入栈的字符:" << c << endl;
            root = createNewNode();
            BinaryNodeptr leftNode ;
            BinaryNodeptr rightNode ;

            root->element = c;
            rightNode = nodeStack.top();
            nodeStack.pop();

            leftNode = nodeStack.top();
            nodeStack.pop();

            root->left = leftNode;
            root->right = rightNode;

            nodeStack.push(root);
            cout << "根结点:" << root->element << endl;
        }
    }
}

void expression_to_tree::printTree(BinaryNodeptr node)
{
    if (node==NULL)
    {
        return;
    }
    else
    {
        printTree(node->left);
        cout << node->element << endl;
        printTree(node->right);
    }
}

expression_to_tree::~expression_to_tree()
{
    deleteAllNode(root);
}

int main(void)
{
    vector<char> str;
    char c;

    BinaryNodeptr tree;

    cout << "Please input expression:" << endl;
    cout << "按q结束输入:" << endl;

    //输入后缀表达式
    c = getchar();
    while (c!='q')
    {
        if (c == '+' || c == '-' || c == '*' || c == '/' || isalnum(c))
        {
            str.push_back(c);
            getchar();
            c = getchar();
        }
        else
        {
            cout << "请输入字母、数字、或运算符" << endl;
            return false;
        }
    }

    expression_to_tree express(str);
    tree = express.root;

    express.printTree(tree);
    
    system("pause");
    return 0;
}

转载于:https://my.oschina.net/u/3397950/blog/1526889