1. 前序遍历
class Solution {
public List<Integer> preorderTraversal(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
preOrder(root, result);
return result;
}
public void preOrder(TreeNode root, ArrayList<Integer> result) {
if (root == null) return;
Stack<TreeNode> stack = new Stack<TreeNode>();
stack.push(root);
while(!stack.isEmpty()) {
TreeNode node = stack.peek();
if(node != null) {
stack.pop();
if (node.right != null) stack.push(node.right);
if (node.left != null) stack.push(node.left);
stack.push(node);
stack.push(null);
}else {
stack.pop();
node = stack.peek();
stack.pop();
result.add(node.val);
}
}
}
}
2. 中序遍历
class Solution {
public List<Integer> inorderTraversal(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
preOrder(root, result);
return result;
}
public void preOrder(TreeNode root, ArrayList<Integer> result) {
if (root == null) return;
Stack<TreeNode> stack = new Stack<TreeNode>();
stack.push(root);
while(!stack.isEmpty()) {
TreeNode node = stack.peek();
if (node != null) {
stack.pop();
if (node.right != null) stack.push(node.right);
stack.push(node);
stack.push(null);
if (node.left != null) stack.push(node.left);
}else {
stack.pop();
node = stack.peek();
stack.pop();
result.add(node.val);
}
}
}
}
3. 后序遍历
class Solution {
public List<Integer> postorderTraversal(TreeNode root) {
ArrayList<Integer> result = new ArrayList<>();
preOrder(root, result);
return result;
}
public void preOrder(TreeNode root, ArrayList<Integer> result) {
if (root == null) return;
Stack<TreeNode> stack = new Stack<TreeNode>();
stack.push(root);
while(!stack.isEmpty()) {
TreeNode node = stack.peek();
if (node != null) {
stack.pop();
stack.push(node);
stack.push(null);
if (node.right != null) stack.push(node.right);
if (node.left != null) stack.push(node.left);
}else {
stack.pop();
node = stack.peek();
stack.pop();
result.add(node.val);
}
}
}
}