【代码随想录算法训练Day14】二叉树的遍历 LeetCode 144.145.94

Day14 二叉树遍历专题

今天堂堂进入二叉树专题，而二叉树中最基础也是最重要的就是二叉树的遍历了。
二叉树的前中后序遍历都是深度优先遍历。

遍历法1.递归

要说遍历哪家好？递归遍历就是吊。
代码简洁大方，实现逻辑清晰易懂。如果还有不清楚的话就用一个二叉树和栈模拟一下，就能领略递归的魅力了。
三种顺序遍历递归的唯一区别就是访问节点语句和递归函数的位置不同，而这也很好记住，访问节点就是中间节点的顺序位置。

LeetCode 144.二叉树的前序遍历

class Solution {
public:

    void preorder(TreeNode* root,vector<int>& res){
        if(root==nullptr) return;
        res.push_back(root->val);
        preorder(root->left,res);
        preorder(root->right,res);
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res;
        preorder(root,res);
        return res;
    }
};

LeetCode 145.二叉树的后序遍历

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

LeetCode 94.二叉树的中序遍历

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

遍历法2.迭代

迭代法相较于递归法，更加直接，但也要更复杂一点。
迭代法的整体思路是利用一个栈来保存遍历过的元素，然后通过改变入栈元素的顺序的方式来达到输出不同的遍历顺序的结果。

LeetCode 144.二叉树的前序遍历

前序遍历的迭代的关键是先输出根节点元素，再将其左右节点压入栈中，由于栈是先进后出，所以要先输出左节点，就要先将右节点压入栈中。

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

LeetCode 145.二叉树的后序遍历

后序遍历巧妙的利用了前序遍历的结果，前序为中左右，而后序为左右中，先将前序遍历中的左右节点入栈顺序调换，这样就变成了中右左，再将序列翻转，就变成了左右中，这种方法实在是巧妙。

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> s;
        s.push(root);
        while(!s.empty()){
            TreeNode* node=s.top();s.pop();
            if(node!=nullptr) res.push_back(node->val);
            else continue;
            s.push(node->left);
            s.push(node->right);
        }
        reverse(res.begin(),res.end());
        return res;
    }
};

LeetCode 94.二叉树的中序遍历

中序遍历的迭代应该是最复杂的了，复杂的点在于我们遍历元素的顺序与输出元素的顺序不一致，这就导致中序的输出很麻烦。
这里使用的是回填的方法，从栈中取出根节点时，并不着急将它放入输出数组中，而是先取出，放入右节点后，再把根节点放回去，这样我们最终输出的思路就与前两种遍历方法有了共通之处。

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

今天学到了很多关于遍历的思维方式，继续加油。