编程导航算法村第七关 | 二叉树的遍历
前序遍历(递归)
public List<Integer> preorderTraversal(TreeNode root) {ArrayList<Integer> result = new ArrayList<Integer>();preorder(root, result);return result;}public void preorder(TreeNode root, List<Integer> result) {if (root == null) {return;}result.add(root.val);preorder(root.left, result);preorder(root.right, result);}
前序遍历(迭代)
- 先迭代到树的最底层,左左端的元素,然后弹出栈,访问他的右节点
public List<Integer> preorderTraversal(TreeNode root) {ArrayList<Integer> result = new ArrayList<Integer>();if (result == null) {return result;}Stack<TreeNode> stack = new Stack<TreeNode>();TreeNode node = root;while (!stack.isEmpty() || node != null) {while (node!= null) {result.add(node.val);stack.push(node);node = node.left;}node = stack.pop();node = node.right;}return result;}
中序遍历(迭代)
public List<Integer> inorderTraversal(TreeNode root) {ArrayList<Integer> result = new ArrayList<>();if (root == null) {return result;}Stack<TreeNode> stack = new Stack<TreeNode>();TreeNode node = root;while (!stack.isEmpty() || node != null) {while (node != null) {stack.push(node);node = node.left;}node = stack.pop();result.add(node.val);node = node.right;}return result;}
后续遍历(反转法)
- 后续遍历相当于在前序遍历的基础上,先访问右节点再访问左节点,最后翻转
public List<Integer> postorderTraversal(TreeNode root) {ArrayList<Integer> result = new ArrayList<>();if (root == null) {return result;}Stack<TreeNode> stack = new Stack<TreeNode>();TreeNode node = root;while (!stack.isEmpty() || node != null) {while (node != null) {stack.push(node);result.add(node.val);node = node.right;}node = stack.pop();node = node.left;}Collections.reverse(result);return result;}