二叉树的遍历（前序中序后序层序）

代码随想录 二叉树的迭代遍历

前序优先遍历

题目

链接: 144. 二叉树的前序遍历

递归

class Solution {
    vector<int> res;
    void preOrder(TreeNode *root)
    {
        if (root)
        {
            res.push_back(root->val);
            preOrder(root->left);
            preOrder(root->right);
        }
    }
public:
    vector<int> preorderTraversal(TreeNode* root) {
        res.clear();
        preOrder(root);
        return res;
    }
};

非递归

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        stack<TreeNode*> stack;
        stack.push(root);
        if (root == nullptr) return {};
        vector<int> res;
        while (!stack.empty())
        {
            TreeNode* node = stack.top();               // 中
            stack.pop();
            res.push_back(node->val);
            // 前序遍历是中左右，每次先处理的是中间节点，那么先将根节点放入栈中，然后将右孩子加入栈，再加入左孩子。
            // 为什么要先加入右孩子，再加入左孩子呢？ 因为这样出栈的时候才是中左右的顺序。
            if (node->right) stack.push(node->right);   // 右（空节点不入栈）
            if (node->left) stack.push(node->left);     // 左（空节点不入栈）
        }
        return res;
    }
};

中序优先遍历

题目

链接: 94. 二叉树的中序遍历

递归

class Solution {
    vector<int> res;
    void inOrder(TreeNode* root)
    {
        if (root)
        {
            inOrder(root->left);
            res.push_back(root->val);
            inOrder(root->right);
        }
     }
public:
    vector<int> inorderTraversal(TreeNode* root) {
        res.clear();
        inOrder(root);
        return res;
    }
};

非递归

中序遍历跟前序遍历和后序遍历都不太一样

中序遍历是左中右，先访问的是二叉树顶部的节点，然后一层一层向下访问，直到到达树左面的最底部，再开始处理节点（也就是在把节点的数值放进result数组中），这就造成了处理顺序和访问顺序是不一致的。

那么在使用迭代法写中序遍历，就需要借用指针的遍历来帮助访问节点，栈则用来处理节点上的元素。

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

后序优先遍历

题目

链接: 145. 二叉树的后序遍历

递归

class Solution {
    vector<int> res;
    void postOrder(TreeNode *root)
    {
        if (root)
        {
            postOrder(root->left);
            postOrder(root->right);
            res.push_back(root->val);
        }
    }
public:
    vector<int> postorderTraversal(TreeNode* root) {
        res.clear();
        postOrder(root);
        return res;
    }
};

非递归

再来看后序遍历，先序遍历是中左右，后序遍历是左右中，那么我们只需要调整一下先序遍历的代码顺序，就变成中右左的遍历顺序，然后在反转result数组，输出的结果顺序就是左右中了

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

此时我们用迭代法写出了二叉树的前后中序遍历，可以看出前序和中序是完全两种代码风格，并不像递归写法那样代码稍做调整，就可以实现前后中序。

这是因为前序遍历中访问节点（遍历节点）和处理节点（将元素放进result数组中）可以同步处理，但是中序就无法做到同步！

层序遍历

代码随想录 二叉树的层序遍历

题目

102. 二叉树的层序遍历

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*>que;
        if (root != nullptr) que.push(root);
        vector<vector<int>> res;
        while (!que.empty())
        {
            int size = que.size();
            vector<int> vec;
            for (int i = 0; i < size; i++)
            {
                TreeNode* node = que.front(); que.pop();
                vec.push_back(node->val);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            res.push_back(vec);
        }
        return res;
    }
};