樹的三種非遞迴遍歷實現
阿新 • • 發佈:2021-06-12
先序:(每個節點左下延伸,沿途依次訪問,且沿途依次將右孩子入棧。不能左下時,讓出棧節點執行前面的過程)
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; } }