理解递归思想:调用自己、压栈、明确终止条件
我的建议是直接把代码背下来
你当然可以尝试去理解,也不是很难想明白,但是不要钻牛角尖了,递归这玩意儿,很不友好
- /**
- * 前序遍历,将结果返回到list中
- *
- * @param root
- * @param res
- */
-
- public static void preOrder(TreeNode root, List
res) { - if (root == null) {
- return;
- }
- res.add(root.val);
- preOrder(root.left, res);
- preOrder(root.right, res);
- }
-
-
- /**
- * 中序遍历,将结果返回到list中
- *
- * @param root
- * @param res
- */
- public static void inOrder(TreeNode root, List
res) { - if (root == null) {
- return;
- }
- inOrder(root.left, res);
- res.add(root.val);
- inOrder(root.right, res);
- }
-
- /**
- * 后序遍历,将结果返回到list中
- *
- * @param root
- * @param res
- */
- public static void postOrder(TreeNode root, List
res) { - if (root == null) {
- return;
- }
- postOrder(root.left, res);
- postOrder(root.right, res);
- res.add(root.val);
- }
- public static List
preOrderTraversal(TreeNode root) { - List
res = new ArrayList(); - if (root == null) {
- return res;
- }
-
- Deque
stack = new LinkedList(); - TreeNode node = root;
- while (!stack.isEmpty() || node != null) {
- while (node != null) {
- res.add(node.val);
- stack.push(node);
- node = node.left;
- }
-
- node = stack.pop();
- node = node.right;
- }
- return res;
- }
- public static List
inorderTraversal(TreeNode root) { - List
res = new ArrayList(); - Deque
stack = new LinkedList(); -
- while (root != null || !stack.isEmpty()) {
- while (root != null) {
- stack.push(root);
- root = root.left;
- }
-
- root = stack.pop();
- res.add(root.val);
- root = root.right;
- }
- return res;
- }
前序遍历和中序遍历的代码不复杂,还是注意:挺不好理解的,不要钻牛角尖,那么我们接下来看下后序遍历
这个相较于前两个更难实现,但可以这样想:
前序遍历的顺序是:中、左、右,我们简单该两个参数,实现遍历顺序为中、右、左
再将遍历结果reserve反转,不就得到左、右、中的遍历顺序了吗?
- public static List
postOrderTraversal(TreeNode root) { - List
res = new ArrayList<>(); - if (root == null) return res;
- Stack
stack = new Stack<>(); - TreeNode node = root;
-
- while (!stack.isEmpty() || node != null) {
- while (node != null) {
- res.add(node.val);
- stack.push(node);
- node = node.right;
- }
- node = stack.pop();
- node = node.left;
- }
- Collections.reverse(res);
- return res;
- }