14代码随想录训练营day14|二叉树part01

花的草

已于 2023-07-18 12:33:38 修改

阅读量544

点赞数 2

文章标签：算法数据结构 c++

于 2023-07-16 21:05:19 首次发布

本文链接：https://blog.csdn.net/m0_62828466/article/details/131753672

版权

本文介绍了如何使用递归和迭代方法遍历二叉树，包括前序遍历、中序遍历和后序遍历。在递归方案中，分别展示了三种遍历的实现代码，而在迭代方案中，对先序、中序和后序遍历进行了详细展示，特别是后序遍历需要额外的反转步骤来得到正确顺序。

摘要由CSDN通过智能技术生成

1、二叉树的递归遍历

1.1、前序遍历

144. 二叉树的前序遍历 - 力扣（LeetCode）

class Solution {
public:
    void ergodic(TreeNode* cur, vector<int>& vec) {
        // 终止条件
        if (cur == NULL) return ;
        // 存根结点
        vec.push_back(cur->val);
        // 递归左子树
        ergodic(cur->left, vec);
        // 递归右子树
        ergodic(cur->right, vec);
    }

    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> result;
        ergodic(root, result);
        return result;
    }
};

1.2、中序遍历

94. 二叉树的中序遍历 - 力扣（LeetCode）

class Solution {
public:
    void ergodic(TreeNode *cur, vector<int>& vec) {
        if (cur == NULL) return ;
        ergodic(cur->left, vec);
        // 对比先序遍历的顺序
        vec.push_back(cur->val);
        ergodic(cur->right, vec);
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> result;
        ergodic(root, result);
        return result;
    }
};

1.3、后序遍历

145. 二叉树的后序遍历 - 力扣（LeetCode）

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

2、迭代法

2.1、先序遍历

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
       // 用栈压入元素
       stack<TreeNode*> mystack;
       vector<int> result;
       if (root == NULL) return result;
       mystack.push(root);
       while (!mystack.empty()) {
           TreeNode *temp = mystack.top();
           result.push_back(temp->val);
           mystack.pop();
           // 先压入right，后压入left，后进先出
           if (temp->right != NULL) mystack.push(temp->right);
           if (temp->left != NULL) mystack.push(temp->left);
       }
       return result;
    }
};

2.2、后序遍历

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        stack<TreeNode*> mystack;
        vector<int> result;
        if (root == NULL) return result;
        mystack.push(root);
        while (!mystack.empty()) {
            TreeNode* temp = mystack.top();
            mystack.pop();
            result.push_back(temp->val);
            if (temp->left != NULL) mystack.push(temp->left);
            if (temp->right != NULL) mystack.push(temp->right);
        }
        // reverse(result.begin(), result.end());
        for (int i = 0, j = result.size() - 1; i < j; i++, j--) {
            swap(result[i], result[j]);
        }
        return result;
    }
};

2.3、中序遍历

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();
                st.pop();
                result.push_back(cur->val);
                cur = cur->right;
            }
        }
        return result;
    }
};