1. 程式人生 > 其它 >樹的三種非遞迴遍歷實現

樹的三種非遞迴遍歷實現

先序:(每個節點左下延伸,沿途依次訪問,且沿途依次將右孩子入棧。不能左下時,讓出棧節點執行前面的過程)

class Solution {
    public List<Integer> preorderTraversal(TreeNode root) {
        LinkedList<Integer> ans=new LinkedList<>();
        LinkedList<TreeNode> sta=new LinkedList<>();
        TreeNode p=root;
        while(p!=null||!sta.isEmpty()){
          while(p!=null){
              ans.add(p.val);//訪問
              sta.add(p.right);//右孩子入棧
              p=p.left;//往左下延伸
          }
          p=sta.removeLast();//讓棧頂也做一次上面的迴圈
        }
    return ans;
    }
}

中序:(每個節點沿左下延伸,沿途節點不訪問併入棧。出棧一個訪問一個,出棧的右孩子執行前面的過程)

class Solution {

    public List<Integer> inorderTraversal(TreeNode root) {
        LinkedList<Integer> ans=new LinkedList<>();
        LinkedList<TreeNode> sta=new LinkedList<>();
        TreeNode p=root;
        while(p!=null||!sta.isEmpty()){
          while(p!=null){
              sta.add(p);//左下延伸,沿途的節點不訪問,入棧
              p=p.left;
          }
          TreeNode node=sta.removeLast();
          ans.add(node.val);
          p=node.right;

        }
        return ans;
    }
}

後序(連結串列的插入和先序不一樣,且入棧的順序相反):


class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        LinkedList<Integer> ans=new LinkedList<>();
        LinkedList<TreeNode> sta=new LinkedList<>();
        TreeNode p=root;
        while(p!=null||!sta.isEmpty()){
            while(p!=null){
                ans.addFirst(p.val);
                sta.add(p.left);
                p=p.right;
            }
            p=sta.removeLast();
        }
        return ans;

    }
}