不递归实现二叉树前序遍历、中序遍历

前序遍历递归实现：

void preorder(TreeNode *root){

        if(!root) return;

        cout<<root->val;        // 前序遍历

        preorder(root->left);

        preorder(root->right);

}

二叉树的前序遍历_牛客题霸_牛客网

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int> res; 
        if(root == NULL)
            return res;
        //辅助栈
        stack<TreeNode*> s; 
        //根节点先进栈
        s.push(root); 
        //直到栈中没有节点
        while(!s.empty()){ 
            //每次栈顶就是访问的元素
            TreeNode* node = s.top(); 
            s.pop();
            res.push_back(node->val);
            //如果右边还有右子节点进栈
            if(node->right) 
                s.push(node->right);
            //如果左边还有左子节点进栈
            if(node->left) 
                s.push(node->left);
        }
        return res;
    }
};

中序遍历递归实现：

void preorder(TreeNode *root){

        if(!root) return;

        preorder(root->left);

        cout<<root->val;        // 中序遍历

        preorder(root->right);

}

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int> res;
        //辅助栈
        stack<TreeNode*> s; 
        //当树节点不为空或栈中有节点时
        while(root != NULL || !s.empty()){ // 二叉树中序遍历就是先下探到最左节点
            //每次找到最左节点
            while(root != NULL){ 
                s.push(root);
                root = root->left;
            } // 收集循环传入的root节点的左子树方向上的所有节点直到最左节点
            TreeNode* node = s.top(); 
            s.pop();
            res.push_back(node->val); // 打印最左节点
            //进入右节点
            root = node->right; // 如果右节点为空则下次循环将弹出栈中当前左节点的父节点
        }
        return res;
    }
};

中序遍历在全局上就是首先在左节点方向下探到最左节点，然后最左节点第一个遍历到其“中序”，予以输出，然后：1、当最左节点有右节点，则接着从右节点开始，在左节点方向下探到最左节点，，，；2、当左节点没有右节点，则从栈中弹出最近的父节点，予以输出