我的做法:递归
/*** Definition for a binary tree node.* public class TreeNode {* int val;* TreeNode left;* TreeNode right;* TreeNode() {}* TreeNode(int val) { this.val = val; }* TreeNode(int val, TreeNode left, TreeNode right) {* this.val = val;* this.left = left;* this.right = right;* }* }*/
class Solution {public List<Integer> inorderTraversal(TreeNode root) {List<Integer> ans = new ArrayList<>();inorder(root, ans);return ans;}public void inorder(TreeNode root, List<Integer> ans) {if (root == null) return;inorder(root.left, ans);ans.add(root.val);inorder(root.right, ans);}
}
但是递归存在效率问题。
栈迭代方法虽然提高了效率,但其嵌套循环却非常烧脑,不易理解,容易造成 “一看就懂,一写就废” 的窘况。
而且对于不同的遍历顺序(前序、中序、后序),循环结构差异很大,更增加了记忆负担。
public static List<Integer> preorderTraversal(TreeNode root) { List<Integer> list = new ArrayList<>(); if (root == null) { return list; } Stack<TreeNode> stack = new Stack<>(); stack.push(root); while (!stack.isEmpty()) { TreeNode node = stack.pop(); list.add(node.val); if (node.right != null) { stack.push(node.right); } if (node.left != null) { stack.push(node.left); } } return list;
}
拓展一种“颜色标记法”
其核心思想如下:
使用颜色标记节点的状态,新节点为白色,已访问的节点为灰色。
如果遇到的节点为白色,则将其标记为灰色,然后将其右子节点、自身、左子节点依次入栈。
如果遇到的节点为灰色,则将节点的值输出。
作者:henry
链接:https://leetcode.cn/problems/binary-tree-inorder-traversal/solutions/25220/yan-se-biao-ji-fa-yi-chong-tong-yong-qie-jian-ming/
来源:力扣(LeetCode) 著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
白色节点可以认为是TreeNode
类型,灰色节点可以认为是Integer
类型,不做额外标记。
class Solution {public List<Integer> inorderTraversal(TreeNode root) {List<Integer> ans = new ArrayList<>();if (root == null) return ans;Stack<Object> st = new Stack<>();st.push(root);while (!st.isEmpty()) {Object o = st.pop();// 逆序入栈,中序遍历就是右、中、左if (o instanceof TreeNode) {st.push(((TreeNode) o).right);st.push(((TreeNode) o).val);st.push(((TreeNode) o).left);} else if (o instanceof Integer) {ans.add((Integer) o);}}return ans;}
}