二叉树的迭代遍历
总结
总体思路是通过栈的入栈出栈顺序来控制输出结果,也是用栈来模拟递归。
如前序遍历,中左右,则根节点,入栈后立即出栈,加入右左两节点。左节点成为下一个根节点。
中序遍历,比较麻烦,左中右,则事先根节点入栈,并不断向左节点探寻,直到叶子节点,然后按顺序出栈中的根节点,每出一个,则继续探寻其右节点是否存在,把右节点当做根节点继续向其左节点探寻。
后序遍历,左右中,事先按照前序遍历的方法反转遍历中右左,时候进行反转。
中序比较麻烦,可以多理解理解。
但是这样迭代中序跟其他两种不一样,写的时候比较麻烦。于是给出了统一解法,打上标记。但个人感觉没什么,理解起来比较麻烦。熟练掌握普通的迭代方法就行。
代码
public static List<Integer> preorderTraversal(TreeNode root){
Stack<TreeNode> tr=new Stack<>();
List<Integer> res=new ArrayList<>();
tr.push(root);
while (!tr.isEmpty()){
TreeNode tempNode=tr.pop();
res.add(tempNode.val);
if (tempNode.right!=null)tr.push(tempNode.right);
if (tempNode.left!=null) tr.push(tempNode.left);
}
return res;
}
public static List<Integer> inorderTraversal(TreeNode root){
Stack<TreeNode> stack=new Stack<>();
List<Integer> res=new ArrayList<>();
stack.push(root);
TreeNode tempNode=root;
while (tempNode != null || !stack.isEmpty()){
if (tempNode != null){
stack.push(tempNode);
tempNode = tempNode.left;
}else{
tempNode = stack.pop();
res.add(tempNode.val);
tempNode = tempNode.right;
}
}
return res;
}
public List<Integer> postorderTraversal(TreeNode root) {
List<Integer> res = new ArrayList<>();
Stack<TreeNode> stack = new Stack<>();
stack.push(root);
while (!stack.isEmpty()){
TreeNode tempNode = stack.pop();
res.add(tempNode.val);
if (tempNode.left != null){
stack.push(tempNode.left);
}
if (tempNode.right != null){
stack.push(tempNode.right);
}
}
Collections.reverse(res);
return res;
}