二叉樹先序,中序和后序遍歷的遞歸和非遞歸實現(xiàn)

遞歸

比較簡單,直接看代碼即可.


    private void preOrderRec(TreeNode node,ArrayList<Integer> pre){
        if(node==null) return;
        pre.add(node.val);
        preOrderRec(node.left,pre);
        preOrderRec(node.right,pre);
    }
    
    private void inOrderRec(TreeNode node,ArrayList<Integer> in){
        if(node==null) return;        
        inOrderRec(node.left,in);
        in.add(node.val);
        inOrderRec(node.right,in);
    }
    
    private void postOrderRec(TreeNode node,ArrayList<Integer> post){
        if(node==null) return; 
        postOrderRec(node.left,post);
        postOrderRec(node.right,post);
        post.add(node.val);
    }

非遞歸

先序遍歷

  1. 申請一個棧,記為s1,將頭結點壓棧.
  2. 每次從棧頂彈出節(jié)點node,打印node的值,如果node的右子節(jié)點不為空,壓棧.如果node的左子結點不為空,壓棧.
  3. 重復步驟2直到棧為空.
//非遞歸版本
    private void preOrder(TreeNode node,ArrayList<Integer> pre){      
       Deque<TreeNode> stack=new ArrayDeque<TreeNode>();
       if(node==null) return;
       stack.push(node);
       while(!stack.isEmpty()){
           node=stack.pop();
           pre.add(node.val);
           if(node.right!=null)stack.push(node.right);
           if(node.left!=null)stack.push(node.left);           
       }
    }

中序遍歷

  1. 申請一個棧,記為s1,當前節(jié)點為node.
  2. 將node及其左邊界壓入棧中,即不斷的利用node=node.left并對其壓棧,然后重復步驟2.
  3. 直到node為空,此時從s1中彈出一個節(jié)點,記為node,打印node的值并讓node=node.right.轉到步驟2.
 private void inOrder(TreeNode node,ArrayList<Integer> in){
        Deque<TreeNode> stack=new ArrayDeque<TreeNode>();
        while(node!=null||!stack.isEmpty()){    
           while(node!=null){
               stack.push(node);
               node=node.left;
           }      
           node=stack.pop();
           in.add(node.val);
           node=node.right;                       
       }        
    }

后序遍歷(兩個棧實現(xiàn))

  1. 申請一個棧,記為s1,然后將頭結點壓入s1.
  2. 從s1中彈出的節(jié)點記為node,先把node的左孩子壓入s1中,再把node的右孩子壓入s1中.
  3. 在整個過程過程中,每個從s1中彈出的節(jié)點都壓入到s2中.
  4. 不斷重復2和3,直到s1為空.
  5. 依次彈出s2的值就是后序遍歷的結果.
 private void postOrder(TreeNode node,ArrayList<Integer> post){
       Deque<TreeNode> stack1=new ArrayDeque<TreeNode>();
       Deque<TreeNode> stack2=new ArrayDeque<TreeNode>();
       if(node==null)return;
       stack1.push(node);
       while(!stack1.isEmpty()){
           node=stack1.pop();
           stack2.push(node);
           if(node.left!=null)stack1.push(node.left);
           if(node.right!=null)stack1.push(node.right);        
       }
       while(!stack2.isEmpty()){
           post.add(stack2.pop().val);
       }
    }
最后編輯于
?著作權歸作者所有,轉載或內(nèi)容合作請聯(lián)系作者
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內(nèi)容