**
二叉樹的非遞歸遍歷
**
中序:
中序遍歷是先左孩子 後跟 最後右孩子的順序。所以應該先一個while循環把所有的根及其左孩子依次入棧,然後一個while循環 將棧頂元素取出,遍歷,如果有右孩子,同樣需要將其及其左孩子入棧。代碼如下
TreeNode tem = root;
LinkedList<TreeNode> stack = new LinkedList<>();
while(tem != null){
stack.push(tem);
tem = tem.left;
}
while(!stack.isEmpty()){
TreeNode node = stack.pop();
//假裝遍歷一下
System.out.print(node.val);
node = node.right;
while (node!=null){
stack.push(node);
node = node.left;
}
}
前序遍歷:
前序遍歷是先跟 然後左右孩子。所以直接將非null根結點入棧,然後一個while循環 將棧頂元素取出 遍歷,如果有右孩子入隊,有左孩子 入隊,代碼如下
if(root!=null){
LinkedList stack = new LinkedList<>();
stack.push(root);
while(!stack.isEmpty()){
TreeNode node = stack.pop();
//假裝遍歷
if(node.right != null){
stack.push(node.right);
}
if(node.left != null){
stack.push(node.left);
}
}
}
後序遍歷:
後序遍歷其實是相對較難的。但是也比較好理解
後序就是先左右孩子然後跟,所以我們任然需要將跟及其左孩子依次入棧,不同的是我們不能直接遍歷,而是peek不是pop拿到到棧頂元素,如果該元素有右孩子並且未被訪問過,那麼我們就將右孩子入棧,並將其左孩子依次入棧。否則 訪問該節點並更新剛剛訪問過的結點到last 變量中(該變量的作用是判斷是否被訪問過的根據,如果一個元素
右孩子爲剛剛訪問過的,那麼就不需要將右孩子及右孩子的左孩子依次入棧了)。代碼如下
List res = new ArrayList<>();
TreeNode last = null;
if(root!=null){
LinkedList<TreeNode> stack = new LinkedList<>();
while(root!=null){
stack.push(root);
root = root.left;
}
while (!stack.isEmpty()){
TreeNode node = stack.peek();
if(node.right != null && last != node.right){
TreeNode tem = node.right;
stack.push(tem);
while(tem.left!=null){
stack.push(tem.left);
tem = tem.left;
}
}else{
TreeNode node1 = stack.pop();
last = node1;
res.add(node1.val);
}
}
}
return res;