二叉树的先序中序后序(非递归)

树的先序、中序、后序(非递归)

先序

class Solution {
public:
    void dfs(TreeNode* root,vector<int>& v)
    {
        stack<TreeNode*>s;
        while(root||!s.empty())
        {
            while(root!=NULL)
            {
                s.push(root);
                v.push_back(root->val);
                root = root->left;
            }
            root = s.top();
            s.pop();
            root = root->right;
        }
    }
    vector<int> preorderTraversal(TreeNode* root) {
        vector<int>v;
        if(root)
            dfs(root,v);
        return v;
    }
};

中序

class Solution {
public:
    void dfs(TreeNode* root,vector<int>& v)
    {
        stack<TreeNode*>s;
        while(root||!s.empty())
        {
            while(root!=NULL)
            {
                s.push(root);
                root = root->left;
            }
            root = s.top();
            s.pop();
            v.push_back(root->val);
            root = root->right;
        }
    }
    vector<int> inorderTraversal(TreeNode* root) {
        vector<int>v;
        if(root)
            dfs(root,v);
        return v;
    }
};

后序

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int>v;
        if(root==NULL)
            return v;
        stack<TreeNode*>s;
        s.push(root);
        while(!s.empty())
        {
            TreeNode* node=s.top();
            s.pop();
            v.push_back(node->val);
            if(node->left!=NULL)
                s.push(node->left);
            if(node->right!=NULL)
                s.push(node->right);
        }
        for(int i=0,j=v.size()-1;i<j;i++,j--)
            swap(v[i],v[j]);
        return v;
    }
};
全部评论

相关推荐

一颗宏心:华为HR晚上过了十二点后还给我法消息。
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务