day-14 代码随想录算法训练营（19）二叉树part1（前中后遍历：递归、迭代）

144.二叉树的前序遍历

感觉递归的思路比较清晰，迭代的话得思考一下

递归：

class Solution {
public:
    void judge(TreeNode*root,vector&res)
    {
        if(root==nullptr)
            return;
        res.push_back(root->val);
        judge(root->left,res);
        judge(root->right,res);
    }
    vector preorderTraversal(TreeNode* root) {
        vectorres;
        judge(root,res);
        return res;

    }
};

迭代：

class Solution {
public:
    
    vector preorderTraversal(TreeNode* root) {
        vectorres;
        stackst;
        if(root==nullptr)
            return res;
        st.push(root);
        while(!st.empty())
        {
            TreeNode*node=st.top();
            st.pop();
            res.push_back(node->val);
            if(node->right) st.push(node->right);
            if(node->left) st.push(node->left);
        }
        return res;

    }
};

94.二叉树的中序遍历

迭代：

class Solution {
public:
    vector inorderTraversal(TreeNode* root) {
        vectorres;
        stackst;
        TreeNode*cur=root;
        while(cur!=nullptr || !st.empty())
        {
            if(cur!=nullptr){//指针访问节点，访问到最底层
                st.push(cur);//将访问的节点放入栈
                cur=cur->left;//左
            }else
            {
                cur=st.top();//从栈里弹出的数据，就是要处理的数据
                st.pop();
                res.push_back(cur->val);//中
                cur=cur->right;//右
            }
        }
        return res;
    }
};

145.二叉树的后序遍历

迭代：

class Solution {
public:
    vector postorderTraversal(TreeNode* root) {
        //思路：后序遍历和前序遍历有小小区别
        vectorres;
        stackst;
        if(root==nullptr)
            return res;
        st.push(root);
        while(!st.empty())
        {
            TreeNode*node=st.top();
            st.pop();
            res.push_back(node->val);

            if(node->left)  
                st.push(node->left);
            if(node->right)
                st.push(node->right);
        }
        reverse(res.begin(),res.end());
        return res;
    }
};