二叉树经典算法题整理

学而时习之。

二叉树遍历系列：

1. 非递归实现二叉树前序遍历。
   思想: 弄一个栈，每次弹出的，就先保存好这个弹出的结果。然后先右后左子树的根节点压入堆栈。

class Solution {
   
public:
    vector<int> preorderTraversal(TreeNode* root) {
   
        vector<int> r;
        
        if (root==nullptr) return r;

        stack<TreeNode*> q;


        q.push(root);

        while (!q.empty())
        {
   
            TreeNode* tmp = q.top();
            q.pop();

            r.push_back(tmp->val);

           
            if (tmp->right) q.push(tmp->right);
            if (tmp->left) q.push(tmp->left);
        }

        
        return r;

    }
    
};

2. 二叉树中序遍历。
   思想：注意到，由于是中序遍历，所以根节点要弹出两次，第一次时，继续压入右左子树；第二次时，保存结点的值。核心是 弄个pair保存访问次数。

class Solution {
   
public:
    vector<int> inorderTraversal(TreeNode* root) {
   
        vector<int> r;
        
        if (!root) return r;

        stack<pair<TreeNode*, int> > s;

        s.push(make_pair(root, 0));


        while (!s.empty())
        {
   
            pair<TreeNode*, int> tmp=s.top();
            s.pop();

            if (tmp.second==1 || ((tmp.first)->right==nullptr && (tmp.first)->left==nullptr)    )
                {
   r.push_back((tmp.first)->val);}

            else if(tmp.second==0)
            {
           
                if ((tmp.first)->right) s.push(make_pair((tmp.first)->right,0));

                tmp.second +=1;
                s.push(tmp);
                
                if ((tmp.first)->left) s.push(make_pair((tmp.first)->left,0));
            }
        }

        return r;



    }

};

3. 二叉树后序遍历。
   思想：与中序是一样的。只不过第一次弹出时，压入堆栈的顺序不一样而已，先自己，然后右边，再左边。

class Solution {
   
public:
    vector<int> postorderTraversal(TreeNode* root) {
   
        vector<int> r;
        if (root==nullptr) return r;

        stack<pair<TreeNode*, int>> s;

        s.push(make_pair(root, 0));

        while (!s.empty())
        {
   
            pair<TreeNode*, int> tmp = s.top();
            s.