代码随想录算法训练营第十四天|144.二叉树的前序遍历 94.二叉树的中序遍历 145.二叉树的后序遍历

LeetCode 144.二叉树的前序遍历

题目链接：144.二叉树的前序遍历

踩坑：先看了视频，所以没踩坑，但是需要注意的是不要陷入回溯法的陷阱。

思路：

1. 递归：（1）确定递归函数的参数与返回值（2）确定返回停止条件（3）确定单层递归逻辑
2. 迭代：用栈来模拟迭代过程。因为遍历的就是要处理的元素，处理完就要弹出，所以在弹出之前就需要把左右孩子都要入栈，同时考虑栈先入后处理的原则，应先压入右孩子，再压入左孩子。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void digui(TreeNode* root, vector<int>& v)
    {
        if(root == nullptr) return;

        v.push_back(root->val);
        digui(root->left, v);
        digui(root->right, v);
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        digui(root, result);
        return result;
    }
};

LeetCode 94.二叉树的中序遍历

题目链接：94.二叉树的中序遍历

踩坑：先看了视频，就没踩什么坑。

思路：

1. 递归：调换单层递归逻辑的顺序即可。
2. 迭代：这里与前序遍历不同，因为每次遍历到的根节点都是“中”，而需要处理的节点是“左”。所以逻辑成了如果当前节点非空，则将左节点压入栈。如果为空，则取栈顶元素处理并将右节点压入栈。这里不需要一次性将左右节点都压入栈，因为遍历的元素不会立马被弹出。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void digui(TreeNode* root, vector<int>& v)
    {
        if(root == nullptr) return;

        digui(root->left, v);
        v.push_back(root->val);
        digui(root->right, v);
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        digui(root, result);
        return result;
    }
};

LeetCode 145.二叉树的后序遍历

题目链接：145. 二叉树的后序遍历

踩坑：无

思路：

1. 递归：调换单层递归逻辑的顺序即可
2. 迭代：通过对前序遍历“中左右”的改动，即先压入左孩子，再压入右孩子，变成“中右左”，再对结果数组反转得到“左右中”即可。

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void digui(TreeNode* root, vector<int>& v)
    {
        if(root == nullptr) return;

        digui(root->left, v);
        digui(root->right, v);
        v.push_back(root->val);
    }
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> result;
        digui(root, result);
        return result;
    }
};