publicList<Integer> preorderTraversal(TreeNode root) {
List<Integer> result =new ArrayList<>();
Stack<TreeNode>stack=newStack<>();
TreeNode p = root;
while(!stack.isEmpty() || p !=null) {
if(p !=null) {
stack.push(p);
result.add(p.val); // Add before going to children
p = p.left;
} else {
TreeNode node =stack.pop();
p = node.right;
}
}
return result;
}
中序遍历
publicList<Integer> inorderTraversal(TreeNode root) {
List<Integer> result =new ArrayList<>();
Stack<TreeNode>stack=newStack<>();
TreeNode p = root;
while(!stack.isEmpty() || p !=null) {
if(p !=null) {
stack.push(p);
p = p.left;
} else {
TreeNode node =stack.pop();
result.add(node.val); // Add after all left children
p = node.right;
}
}
return result;
}
后序遍历
publicList<Integer> postorderTraversal(TreeNode root) {
LinkedList<Integer> result =new LinkedList<>();
Stack<TreeNode>stack=newStack<>();
TreeNode p = root;
while(!stack.isEmpty() || p !=null) {
if(p !=null) {
stack.push(p);
result.addFirst(p.val); // Reverse the process of preorder
p = p.right; // Reverse the process of preorder
} else {
TreeNode node =stack.pop();
p = node.left; // Reverse the process of preorder
}
}
return result;
}