树的非递归遍历

最新推荐文章于 2021-09-12 14:48:58 发布

原创最新推荐文章于 2021-09-12 14:48:58 发布 · 397 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树的非递归遍历

算法同时被 2 个专栏收录

28 篇文章

订阅专栏

笔试

17 篇文章

订阅专栏

本文介绍了二叉树的三种遍历方法：先序、中序和后序遍历，并提供了详细的C++实现代码。先序遍历首先访问根节点，然后遍历左子树，最后遍历右子树；中序遍历则先遍历左子树，再访问根节点，最后遍历右子树；后序遍历先遍历左右子树，最后访问根节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.先序（我只把这些当模板）

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

2.中序

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

3.后序

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> res;
        if(root==NULL)return res;
        stack<TreeNode*> ms;
        TreeNode *p=root,*pre=NULL;
        while(!ms.empty()||p)
        {
            if(p)
            {
                ms.push(p);
                p=p->left;
            }
            else
            {
                p=ms.top();
                if(p->right!=NULL&&p->right!=pre)p=p->right;
                else 
                {
                    res.push_back(p->val);
                    ms.pop();
                    pre=p;
                    p=NULL;
                }
            }
        }
        return res;
    }
};

看看大神的代码（以下代码来源于leetcode）

1.先序（看起来很优美）

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

2.中序( o(1)空间复杂度）

Morris traversal

public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class Solution {
    public List<Integer> inorderTraversal(TreeNode root) {
        if(root == null) return new ArrayList<Integer>();
        List<Integer> res = new ArrayList<Integer>();
        TreeNode pre = null;
        while(root != null){
        	if(root.left == null){
        		res.add(root.val);
        		root = root.right;
        	}else{
        		pre = root.left;
        		while(pre.right != null && pre.right != root){
        			pre = pre.right;
        		}
        		if(pre.right == null){
        			pre.right = root;
        			root = root.left;
        		}else{
        			pre.right = null;
        			res.add(root.val);
        			root = root.right;
        		}
        	}
        }
        return res;
    }
}

3.后序（美得不敢直视）

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