美文网首页
关于二叉树的三种遍历的非递归实现

关于二叉树的三种遍历的非递归实现

作者: yangqi916 | 来源:发表于2017-01-12 21:38 被阅读0次

    preorder and inorder

      1. 思路:先从root开始一路沿着left把节点push到stack里,读出stack里的node,再移向node->right, 具体访问val的时机改一下就可得到pre 和 in 两个order了。
    • 2.preorder

    
    // preorder
    vector<int> dfs(TreeNode* root) {
            TreeNode* p = root;
            stack<TreeNode*>s;
            vector<int>res;
            
            while (p != NULL || !s.empty()) {
                while (p) {
                    res.push_back(p->val);
                    s.push(p);
                    p = p->left;
                }
                
                p = s.top();
                s.pop();
                p = p->right;
            }
            
            return res;
        }
    
    • 3.inorder
    
    // preorder
    vector<int> dfs(TreeNode* root) {
            TreeNode* p = root;
            stack<TreeNode*>s;
            vector<int>res;
            
            while (p != NULL || !s.empty()) {
                while (p) {
                    s.push(p);
                    p = p->left;
                }
                
                p = s.top();
                s.pop();
                res.push_back(p->val);
                p = p->right;
            }
            
            return res;
        }
    

    postorder

      1. 思路:比起上述情况,我们需要在出栈的时候多判断其右节点是否已经访问过了,如果是的,则可以访问,否则,先加入右边的,其他相同。先从root开始一路沿着left把节点push到stack里,读出stack里的node,再移向node->right。
    前序、中序、后序的非递归遍历中,要数后序最为麻烦,如果只在栈中保留指向结点的指针,那是不够的,必须有一些额外的信息存放在栈中。
    方法有很多,这里只举一种,先定义栈结点的数据结构
    typedef struct{Node * p; int rvisited;}SNode //Node 是二叉树的结点结构,rvisited==1代表p所指向的结点的右结点已被访问过。
    
    lastOrderTraverse(BiTree bt){
      //首先,从根节点开始,往左下方走,一直走到头,将路径上的每一个结点入栈。
      p = bt;
      while(bt){
        push(bt, 0); //push到栈中两个信息,一是结点指针,一是其右结点是否被访问过
        bt = bt.lchild;
      }
    
      //然后进入循环体
      while(!Stack.empty()){ //只要栈非空
        sn = Stack.getTop(); // sn是栈顶结点
    
        //注意,任意一个结点N,只要他有左孩子,则在N入栈之后,N的左孩子必然也跟着入栈了(这个体现在算法的后半部分),所以当我们拿到栈顶元素的时候,可以确信这个元素要么没有左孩子,要么其左孩子已经被访问过,所以此时我们就不关心它的左孩子了,我们只关心其右孩子。
    
        //若其右孩子已经被访问过,或是该元素没有右孩子,则由后序遍历的定义,此时可以visit这个结点了。
        if(!sn.p.rchild || sn.rvisited){
          p = pop();
          visit(p);
        }
        else //若它的右孩子存在且rvisited为0,说明以前还没有动过它的右孩子,于是就去处理一下其右孩子。
        { 
          //此时我们要从其右孩子结点开始一直往左下方走,直至走到尽头,将这条路径上的所有结点都入栈。
    
          //当然,入栈之前要先将该结点的rvisited设成1,因为其右孩子的入栈意味着它的右孩子必将先于它被访问(这很好理解,因为我们总是从栈顶取出元素来进行visit)。由此可知,下一次该元素再处于栈顶时,其右孩子必然已被visit过了,所以此处可以将rvisited设置为1。
          sn.rvisited = 1;
    
          //往左下方走到尽头,将路径上所有元素入栈
          p = sn.p.rchild;
          while(p != 0){
            push(p, 0);
            p = p.lchild;
          }
        }//这一轮循环已结束,刚刚入栈的那些结点我们不必管它了,下一轮循环会将这些结点照顾的很好。
      }
    }
    

    相关文章

      网友评论

          本文标题:关于二叉树的三种遍历的非递归实现

          本文链接:https://www.haomeiwen.com/subject/fwjjbttx.html