递归与栈（二叉树前中后遍历）

 写法一：递归

class Solution {
public:
    void preorder(TreeNode* root, vector<int> &res) {
        if (root == NULL) {
            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;
    }
};

写法二：栈

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

通过这两种方法可以发现，递归的底层其实是栈，递归的过程就是每次递归会把函数的变量，参数值，返回地址压到栈中，等递归返回的时候，再把栈顶弹出去。感觉就很奇妙！

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

 中序遍历有点特殊，需要一个指针，先指到最左端，当cur为null时，让cur为栈顶元素然后输出其值，相当于已经到了“中”这个地步，然后向右走一格。

 

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

其实后序遍历就是改了一下先序遍历，因为先序遍历是中左右，我先把他变为中右左，然后再把他反转，变为左右中，则是后序遍历。