LeetCode树的遍历

因为刷题遇见很多次树的遍历，特此记录一下

递归写法

递归写法比较简短和简单，只以先序遍历为例：

preorder

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

迭代写法

preorder

第一种

 vector<int> preorderTraversal(TreeNode* root) {
        if (!root) return {};
        vector<int> res;
        stack<TreeNode*> s{{root}};
        while (!s.empty()) {
            TreeNode *t = s.top(); s.pop();
            res.push_back(t->val);
            if (t->right) s.push(t->right);
            if (t->left) s.push(t->left);
        }
        return res;
    }

第二种

 vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res;
        stack<TreeNode*> st;
        TreeNode *p = root;
        while (!st.empty() || p) {
            if (p) {
                st.push(p);
                res.push_back(p->val);
                p = p->left;
            } else {
                p = st.top(); st.pop();
                p = p->right;
            }
        }
        return res;
    }

inorder

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

postorder

后序遍历的顺序是左-右-根，而先序遍历的顺序是根-左-右，二者其实还是很相近的，可以先在先序遍历的方法上做些小改动，使其遍历顺序变为根-右-左，然后翻转一下，就是左-右-根啦

 vector<int> postorderTraversal(TreeNode* root) {
        if (!root) return {};
        vector<int> res;
        stack<TreeNode*> s{{root}};
        while (!s.empty()) {
            TreeNode *t = s.top(); s.pop();
            res.insert(res.begin(), t->val);
            if (t->left) s.push(t->left);
            if (t->right) s.push(t->right);
        }
        return res;
    }

Acknowledge

LeetCode144先序遍历
LeetCode145后序遍历
LeetCode94 中序遍历