代码随想录算法训练营第十四天|LeetCode144 二叉树的前序遍历、LeetCode145 二叉树的后序遍历、LeetCode94 二叉树的中序遍历

题1：

指路：LeetCode144 二叉树的前序遍历

思路与代码：

1.递归

这一块需要掌握二叉树和递归的知识，由于我个人二叉树掌握的还可以所以在这里不另外花时间总结。关于递归遍历，我们需要确定递归参数和返回值，终止条件和单层递归逻辑。在这个题前序遍历节点无返回值，因此类型为void；终止条件为节点值为空；单层递归逻辑为二叉树的前序遍历。代码如下：

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

2.迭代

 

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;
    }
};

题2：

指路：LeetCode145 二叉树的后序遍历

思路与代码：

1.递归

与前序遍历大同小异，理解前序了后序问题也不大。代码如下：

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

2.迭代1

class Solution {
public:
    vector<int> postorderTraversal(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->left) st.push(node->left);
        if (node->right) st.push(node->right);
    }
    reverse(result.begin(), result.end());
    return result;
    }
};

题3：

指路：LeetCode94 二叉树的中序遍历

思路与代码：

1.递归

依旧没什么好说的。代码如下：

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

2.迭代

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;
    }
};