LeetCode | C++ 144.二叉树的前序遍历、145.二叉树的后序遍历、94.二叉树的中序遍历

该知识点参考卡哥代码随想录内容

• 二叉树可以用数组按顺序存储，也可以用指针进行链式存储。
• 数组存储方式：如果父节点的数组下标是 i，那么它的左孩子就是 i * 2 + 1，右孩子就是 i * 2 + 2。
• 但用链式表示，更易理解，故一般使用链式存储二叉树。

递归三要素：

• 确定递归函数的参数和返回值
• 确定终止条件
• 确定单层递归的逻辑

144.二叉树的前序遍历

递归写法

class Solution {
public:
    // 递归写法
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root, result);
        return result;
    }
    void traversal(TreeNode* cur, vector<int>& result) {
        if (cur == NULL) return;
        result.push_back(cur->val);
        traversal(cur->left, result);
        traversal(cur->right, result);
    }
};

迭代写法

class Solution {
public:
    // 迭代写法：
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode *> stack1;
        TreeNode* node;
        stack1.push(root);
        while (!stack1.empty()) {
            node = stack1.top();
            stack1.pop();
            if (node != NULL) {
                result.push_back(node->val);
            } else continue;
            stack1.push(node->right);
            stack1.push(node->left);
        }
        return result;
    }
};

145.二叉树的后序遍历

递归写法

class Solution {
public:
    // 递归写法
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root, result);
        return result;
    }
    void traversal(TreeNode* cur, vector<int>& result) {
        if (cur == NULL) return;
        traversal(cur->left, result);
        traversal(cur->right, result);
        result.push_back(cur->val);
    }
};

迭代写法

• 后序迭代写法基于前序迭代写法，将中左右 先改为 中右左， 然后进行倒序排列，双指针法

class Solution {
public:
    // 迭代写法：基于前序迭代写法
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> stack1;
        TreeNode* node;
        stack1.push(root);
        while (!stack1.empty()) {
            node = stack1.top();
            stack1.pop();
            if (node != NULL) {
                result.push_back(node->val);
            } else continue;
            stack1.push(node->left);
            stack1.push(node->right);
        }
        reverse(result.begin(), result.end());
        return result;
    }
};

94.二叉树的中序遍历

递归写法

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        traversal(root, result);
        return result;
    }
    void traversal(TreeNode* cur, vector<int>& result) {
        if (cur == NULL) return;
        traversal(cur->left, result);
        result.push_back(cur->val);
        traversal(cur->right, result);
    }
};

迭代写法

• 对于中序遍历来说，处理的顺序和遍历的顺序是不一致的，所以需要重新写
• 我们需要一个指针，来帮助我们去遍历这个二叉树，对于二叉树而言，先访问的是根节点，我们遍历的时候，要用栈来记录遍历过的顺序，因为栈是按照遍历后的顺序进行逆向输出的
• 直到遍历到最左下的一个节点，此时，我们把该节点当作是一颗独立的二叉树，继续按照左中右的顺序。
  整体思路：用指针来遍历节点、用栈来记录我们遍历过的节点，然后再从栈里弹出元素。

class Solution {
public:
    // 迭代法
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        stack<TreeNode*> stack1;
        TreeNode* node = root;
       
        while (node != NULL || !stack1.empty()) {
            if (node != NULL) {
                stack1.push(node);
                node =  node->left;
            } else {
                node = stack1.top();
                stack1.pop();
                result.push_back(node->val);
                node = node->right;
            }
        }
        return result;
    }
};