二叉树的基本操作实现

1. 二叉树结点结构定义如下：

// 树节点定义
struct TreeNode {
    int data;
    TreeNode* lChild;
    TreeNode* rChild;
};

2. 二叉树的基本操作函数如下：

• TreeNode* createBinTree(); // 先序创建二叉树（递归）
• void preOrderTraverse(TreeNode* head); // 先序遍历（递归）
• void inOrderTraverse(TreeNode* head); // 中序遍历（递归）
• void postOrderTraverse(TreeNode* head); // 后序遍历（递归）
• void preOrderTraverse_1(TreeNode* head); // 先序遍历（非递归）
• void inOrderTraverse_1(TreeNode* head); // 中序遍历（非递归）
• void postOrderTraverse_1(TreeNode* head); // 后序遍历（非递归）

3. 具体代码实现如下：

#include <iostream>
#include <stack>

using namespace std;

// 树节点定义
struct TreeNode {
    int data;
    TreeNode* lChild;
    TreeNode* rChild;
};

// 先序创建二叉树（递归）
TreeNode* createBinTree() {
    TreeNode* head = NULL;
    int val = 0;
    cin >> val;
    if (val != 0) {
        head = new TreeNode;
        head->data = val;
        head->lChild = createBinTree();
        head->rChild = createBinTree();
    }
    return head;
}

// 先序遍历（递归）
void preOrderTraverse(TreeNode* head) {
    if (head) {
        cout << head->data << " ";
        preOrderTraverse(head->lChild);
        preOrderTraverse(head->rChild);
    }
}

// 中序遍历（递归）
void inOrderTraverse(TreeNode* head) {
    if (head) {
        inOrderTraverse(head->lChild);
        cout << head->data << " ";
        inOrderTraverse(head->rChild);
    }
}

// 后序遍历（递归）
void postOrderTraverse(TreeNode* head) {
    if (head) {
        postOrderTraverse(head->lChild);
        postOrderTraverse(head->rChild);
        cout << head->data << " ";
    }
}

// 先序遍历（非递归）
void preOrderTraverse_1(TreeNode* head) {
    TreeNode* node = head;
    stack<TreeNode*> nodeStack;
    while (node != NULL || !nodeStack.empty()) {
        while (node != NULL) {
            cout << node->data << " ";
            nodeStack.push(node);
            node = node->lChild;
        }
        if (!nodeStack.empty()) {
            node = nodeStack.top();
            nodeStack.pop();
            node = node->rChild;
        }
    }
}

// 中序遍历（非递归）
void inOrderTraverse_1(TreeNode* head) {
    TreeNode* node = head;
    stack<TreeNode*> nodeStack;
    while (node != NULL || !nodeStack.empty()) {
        while (node != NULL) {
            nodeStack.push(node);
            node = node->lChild;
        }
        if (!nodeStack.empty()) {
            node = nodeStack.top();
            nodeStack.pop();
            cout << node->data << " ";
            node = node->rChild;
        }
    }
}

// 后序遍历（非递归）
void postOrderTraverse_1(TreeNode* head) {
    // 待续。。
}

int main() {

    TreeNode* head = NULL;
    head = createBinTree();

    cout << "preOrderTraverse: " << endl;
    preOrderTraverse(head);
    cout << endl << "inOrderTraverse: " << endl;
    inOrderTraverse(head);
    cout << endl << "postOrderTraverse: " << endl;
    postOrderTraverse(head);

    cout << endl << "preOrderTraverse_1: " << endl;
    preOrderTraverse_1(head);
    cout << endl << "inOrderTraverse_1: " << endl;
    inOrderTraverse_1(head);
    cout << endl;

    system("pause");

    return 0;
}

4. 运行结果截图如下：

转载于:https://www.cnblogs.com/yiluyisha/p/9269024.html