代码随想录算法训练营第十四天 | 二叉树理论基础，递归遍历，迭代遍历，统一迭代

堆是什么？

堆被定义为具有特殊有序性质的树状结构。在最大堆中，对于任意父节点，其值大于或等于其子节点的值。在最小堆中，对于任意父节点，其值小于或等于其子节点的值。一般来讲，堆是一棵完全二叉树

在插入和删除元素时，堆具有O(logn)的时间复杂度，查找堆顶元素有O(1)的时间复杂度，非常快速

二叉树的遍历方式

二叉树主要有两种遍历方式：

1. 深度优先遍历：先往深走，遇到叶子节点再往回走。根据中间节点的遍历顺序可以分为前序、中序、后序遍历三种遍历方式。
2. 广度优先遍历：一层一层的去遍历。

//深度遍历的三种方式

#include <iostream>

// 二叉树的结点结构
struct TreeNode {
    int value;
    TreeNode* left;
    TreeNode* right;

    TreeNode(int val) : value(val), left(nullptr), right(nullptr) {}
};

// 前序遍历
void preorderTraversal(TreeNode* root) {
    if (root == nullptr) {
        return;
    }

    std::cout << root->value << " "; // 先访问根节点
    preorderTraversal(root->left);   // 递归遍历左子树
    preorderTraversal(root->right);  // 递归遍历右子树
}

// 中序遍历
void inorderTraversal(TreeNode* root) {
    if (root == nullptr) {
        return;
    }

    inorderTraversal(root->left);   // 递归遍历左子树
    std::cout << root->value << " "; // 先访问根节点
    inorderTraversal(root->right);  // 递归遍历右子树
}

// 后序遍历
void postorderTraversal(TreeNode* root) {
    if (root == nullptr) {
        return;
    }

    postorderTraversal(root->left);   // 递归遍历左子树
    postorderTraversal(root->right);  // 递归遍历右子树
    std::cout << root->value << " ";  // 先访问根节点
}

int main() {
    // 创建一棵二叉树
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);

    std::cout << "前序遍历结果：";
    preorderTraversal(root);
    std::cout << std::endl;

    std::cout << "中序遍历结果：";
    inorderTraversal(root);
    std::cout << std::endl;

    std::cout << "后序遍历结果：";
    postorderTraversal(root);
    std::cout << std::endl;

    return 0;
}

//广度优先遍历

#include <iostream>
#include <queue>

struct TreeNode {
    int value;
    TreeNode* left;
    TreeNode* right;

    TreeNode(int val) : value(val), left(nullptr), right(nullptr) {}
};

void breadthFirstTraversal(TreeNode* root) {
    if (root == nullptr) {
        return;
    }

    std::queue<TreeNode*> q;
    q.push(root);

    while (!q.empty()) {
        TreeNode* current = q.front();
        q.pop();
        std::cout << current->value << " ";

        if (current->left) {
            q.push(current->left);
        }
        if (current->right) {
            q.push(current->right);
        }
    }
}

int main() {
    TreeNode* root = new TreeNode(1);
    root->left = new TreeNode(2);
    root->right = new TreeNode(3);
    root->left->left = new TreeNode(4);
    root->left->right = new TreeNode(5);

    std::cout << "广度优先遍历结果：";
    breadthFirstTraversal(root);
    std::cout << std::endl;

    return 0;
}

二叉树的递归遍历

代码在上面已经写过了，三种不同的遍历方式没有本质上的区别

迭代遍历

迭代法使用循环结构（通常是栈或队列）来模拟递归过程，从而遍历二叉树，它的优点是可以有效地避免递归的深度过大导致的栈溢出问题，适用于大型二叉树，但是具有需要手动管理内存空间以及代码更加复杂的问题

和递归遍历相同的是迭代遍历也有三种方式，原则是什么序遍历就在处理对应节点的位置进行数据输出，最简单的是前序遍历，因为我们需要的是根节点并且需要输出的第一个节点也是根节点，代码如下：

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> st;
        vector<int> result;
        if (root == NULL) return result;
        st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();                       // 中
            st.pop();
            result.push_back(node->val);
            if (node->right) st.push(node->right);           // 右（空节点不入栈）
            if (node->left) st.push(node->left);             // 左（空节点不入栈）
        }
        return result;
    }
};

其中先入栈的是右节点，因为栈是FILO，左节点先出栈应该后入栈，对于后序遍历，只需要将“左右中”的顺序倒过来变成“中右左” ，在前序遍历“中左右”的基础上交换左右节点顺序再倒序输出即可

中序遍历相对复杂，因为此时元素的处理顺序和访问顺序不一致，此时使用指针辅助访问节点

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while (cur != NULL || !st.empty()) {
            if (cur != NULL) { // 指针来访问节点，访问到最底层
                st.push(cur); // 将访问的节点放进栈
                cur = cur->left;                // 左
            } else {
                cur = st.top(); // 从栈里弹出的数据，就是要处理的数据（放进result数组里的数据）
                st.pop();
                result.push_back(cur->val);     // 中
                cur = cur->right;               // 右
            }
        }
        return result;
    }
};

统一迭代

统一迭代的代码比较复杂，但是相较上一种迭代的写法，更换代码顺序就可以写出三种遍历方式，编写的规则可以总结为在中节点访问后加入空节点做标记，从而改变总的访问顺序，只允许在访问到空节点后对应的数据出栈，。

以中序为例，从根节点开始向下访问到底部节点并按“右中左”的顺序入栈（反向出栈），其中所有的中节点前都插入一个NULL，也就是除了最底部的节点以外都按顺序排好了，并且都有NULL标记（相对于自己的左右节点而言大家都是中节点），对于底部的节点而言，因为没有字节点，自己最后也会被判定为一个没有左右节点的中节点，并且按照我们需要的顺序排列好了，剩下的工作就是出栈

//中序统一迭代遍历
class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> st;
        if (root != NULL) st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();
            if (node != NULL) {
                st.pop(); // 将该节点弹出，避免重复操作，下面再将右中左节点添加到栈中
                if (node->right) st.push(node->right);  // 添加右节点（空节点不入栈）

                st.push(node);                          // 添加中节点
                st.push(NULL); // 中节点访问过，但是还没有处理，加入空节点做为标记。

                if (node->left) st.push(node->left);    // 添加左节点（空节点不入栈）
            } else { // 只有遇到空节点的时候，才将下一个节点放进结果集
                st.pop();           // 将空节点弹出
                node = st.top();    // 重新取出栈中元素
                st.pop();
                result.push_back(node->val); // 加入到结果集
            }
        }
        return result;
    }
};