二叉树的遍历汇总,刷题必备

   日期:2020-05-18     浏览:85    评论:0    
核心提示:1.简介二叉树的遍历主要分为深度优先遍历和广度优先遍历两大类,其中深度优先遍历又分为前序遍历、中序遍历、后序遍历。我们以下面这棵树来具体说明这些遍历之间的区别。 广度优先遍历:从左到右依次遍历每一层,即按1 2 3 4 5 6顺序遍历。前序遍历:先遍历父节点,再遍历左子树,最后遍历右...数据结构与算法

1.简介

二叉树的遍历主要分为深度优先遍历和广度优先遍历两大类,广度优先遍历是按层从上到下遍历(先遍历第一层,再遍历第二层...),深度优先遍历则是每一次按树的一条路径遍历至叶子节点,然后遍历另外路径上的节点。
其中深度优先遍历又分为前序遍历、中序遍历、后序遍历。我们以下面这棵树来具体说明这些遍历之间的区别。

                                                  

广度优先遍历:从左到右依次遍历每一层,即按1 2 3 4 5 6顺序遍历。

前序遍历:先遍历父节点,再遍历左子树,最后遍历右子树,即按1 2 4 5 3 6顺序遍历。

中序遍历:先遍历左子树,再遍历父节点,最后遍历右子数,即按4 2 5 1 6 3顺序遍历。

 后序遍历:先遍历左子树,再遍历右子树,最后遍历父节点,即按4 5 2 6 3 1顺序遍历。

此外,还有常数级空间的morris遍历,更为复杂,在我的另一篇博客有介绍。

2.代码实现

上面几种遍历的实现都可以分为递归和迭代两种方法。

2.1广度优先遍历

递归法

为了区分层次这里输出二维数组,使用level记录当前遍历层次。第i层的数据就放入v[i-1]中。

    void search(TreeNode* root,vector<vector<int>>&v,int level)
    {
        if(!root)return;
        if(v.size()==level)v.resize(level+1);
        v[level].emplace_back(root->val);
        search(root->left,v,level+1);
        search(root->right,v,level+1);
    }
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        vector<vector<int>>v;
        search(root,v,0);
        return v;
    }

迭代法

使用队列queue

    vector<vector<int>> levelOrder(Node* root) {
        vector<vector<int>>v;
        if(!root)return v;
        queue<Node*>tmp;
        tmp.push(root);
        while(!tmp.empty())
        {
            int size=tmp.size();
            v.push_back({});
            while(size--)
            {
                Node* cur=tmp.front();
                tmp.pop();
                v.back().push_back(cur->val);
                tmp.push(cur->left);
                tmp.push(cur->right);
            }
        }
        return v;
    }

2.2前序遍历

递归法

vector<int> results;
vector<int> inorderTraversal(TreeNode* root) {
    traverse(root);
    return results;
}

void traverse(TreeNode* node){
    if(node==NULL) return;
    results.push_back(node->val);
    traverse(node->left); 
    traverse(node->right);
}

迭代法

思路:每到一个节点 A,就应该立即访问它。因为,每棵子树都先访问其根节点。对节点的左右子树来说,也一定是先访问根。
在 A 的两棵子树中,遍历完左子树后,再遍历右子树。因此,在访问完根节点后,遍历左子树前,要将右子树压入栈。

vector<int> preorderTraversal(TreeNode* root) {
    vector<int> ans;
    if(root==NULL) return ans;
    stack<TreeNode*> s;
    s.push(root);

    while(!s.empty()){
        TreeNode* top=s.top();
        s.pop();
        ans.emplace_back(top->val);
        if(top->right) s.push(top->right);
        if(top->left) s.push(top->left);
    }
    return ans;
}

2.3中序遍历

递归法

vector<int> results;
vector<int> inorderTraversal(TreeNode* root) {
    traverse(root);
    return results;
}

void traverse(TreeNode* node){
    if(node==NULL) return;
    traverse(node->left);
    results.push_back(node->val);
    traverse(node->right);
}

迭代法

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

    return results;
}

2.4后序遍历

递归法

vector<int> results;
vector<int> inorderTraversal(TreeNode* root) {
    traverse(root);
    return results;
}

void traverse(TreeNode* node){
    if(node==NULL) return;
    traverse(node->left);
    traverse(node->right);
    results.push_back(node->val);
}

迭代法

迭代法实现可以有两种方法,一种就是仿照前序遍历的方法,按中右左进行遍历,然后进行逆序,就得到了左右中的后序遍历结果。实现可以参考前序遍历,这里不再多写。

不使用反转的后序遍历比前面两种复杂,因为如果将父节点和右节点放入栈中,父节点和右节点无法区分。思路如下:

  • 先遍历左节点直到左节点为null。
  • 开始遍历右节点,若该右节点有左节点,优先遍历左节点。
  • 使用rightchild来记录右节点是否已被遍历过。若是:则说明以该点为根的子树已被遍历,输出根节点。若否:就开始遍历右节点,回到第二步。
    vector<int> postorderTraversal(TreeNode* root) {
        vector<int> ans;
        stack<TreeNode*> stk;
        TreeNode* cur = root;
        TreeNode* rightchild = NULL;
        while(cur || !stk.empty()){
            while(cur != NULL){
                stk.push(cur);
                cur = cur -> left;
            }
            cur = stk.top();
            if(!cur -> right|| rightchild == cur -> right){
                ans.push_back(cur -> val);
                stk.pop();
                rightchild = cur;
                cur = NULL;
            }
            else{
                rightchild = NULL;
                cur = cur -> right;
            }
        }
        return ans;
    }

 

 
打赏
 本文转载自:网络 
所有权利归属于原作者,如文章来源标示错误或侵犯了您的权利请联系微信13520258486
更多>最近资讯中心
更多>最新资讯中心
0相关评论

推荐图文
推荐资讯中心
点击排行
最新信息
新手指南
采购商服务
供应商服务
交易安全
关注我们
手机网站:
新浪微博:
微信关注:

13520258486

周一至周五 9:00-18:00
(其他时间联系在线客服)

24小时在线客服