代码随想录算法训练营第十四天 | 二叉树的遍历

不知不觉参加代码随想录的算法训练营已经两周了，今天进度也是到达了树。

今天是基础的二叉树的遍历，分为前序遍历，中序遍历，后序遍历。

方法主要有两种，第一种是递归遍历，第二种是迭代遍历。

递归遍历

144.二叉树的前序遍历

145.二叉树的后序遍历

94.二叉树的中序遍历

递归遍历的模板

res.push_back(cur->val);   //中
traversal(cur->left, res); //左
traversal(cur->right, res);//右

思路

上面三句分别代表中，左，右。

前序遍历是中左右，中序遍历是左中右，后序遍历是左右中。

根据遍历顺序，调整上面三句的顺序就好。

代码实现

这里给出前序遍历的代码，中序遍历与后续遍历，相应的变化就好

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

收获

 传递数组时，记得加&。

迭代遍历

使用完递归遍历之后，我们再来尝试迭代遍历。

迭代遍历就是使用栈。但是具体嘛......

前序遍历

力扣题目链接

前序遍历的顺序是中左右，那么先处理中间节点，然后将中间节点放入栈中，然后将右孩子放入栈中，然后将左孩子放入栈中。

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> stk;
        vector<int> res;
        if (root == NULL) return res;
        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;
    }
};

注意

在入栈时，一定记得要先检查该节点是否存在。

后序遍历

力扣题目链接

后序遍历的顺序是左右中，那么只需要先将中间节点入栈，然后将左孩子入栈，然后右孩子入栈。最后将整个结果反转过来，就是后序遍历了。

代码实现

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> stk;
        vector<int> res;
        if (root == NULL) return res;
        st.push(root);
        while (!st.empty()) {
            TreeNode* node = st.top();                     
            st.pop();
            result.push_back(node->val);
            if (node->left) st.push(node->left); 
            if (node->right) st.push(node->right);          
        }
        return result;
    }
};

中序遍历

力扣题目链接

中序遍历的迭代代码就没法像前序和后续一样修改一点点就完成了。

思路

使用迭代进行中序遍历时，不断将左子树入栈，然后进行处理节点。

代码实现

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int>res;
        stack<TreeNode*>stk;
        TreeNode* cur = root;
        while(cur!= nullptr||!stk.empty())
        {
            while(cur!= nullptr)
            {
                stk.push(cur);
                cur = cur->left;
            }
            cur = stk.top();
            stk.pop();
            res.push_back(cur->val);
            cur = cur->right;
        }
        return res;
    }
};