二叉的前向遍历,即先输出当且节点的值,再遍历左子树,右子树。
同样的,先来造一个二叉树。
我先跟大家讲一个傻瓜式的做法(一开始我是这么做的,但发现速度上不行)
1.先输出当前节点的值,即4。
2.接着右子树入栈,左子树入栈。
3. 当前节点 = pop() ,重复步骤1,2.
代码:
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new LinkedList<>();
Stack<TreeNode> store = new Stack<>();
if (root == null) {
return list;
}
store.push(root);
while (!store.empty()) {
TreeNode cur = store.pop();
list.add(cur.val);
if (cur.right != null) {
store.push(cur.right);
}
if (cur.left != null) {
store.push(cur.left);
}
}
return list;
}
性能:
发现这个解法的问题了吗?
右子树是可以存储起来,但左子树马上就要用了。为什么还要存储起来呢?(其实时间就浪费在左子树的无效的入栈,出栈上)
更新版本:
碰到有右子树的,就只将右子树入栈。
改进版性能:
改进版代码:
public List<Integer> preorderTraversal(TreeNode root) {
List<Integer> list = new LinkedList<>();
Stack<TreeNode> store = new Stack<>();
putLeftMost(list, store, root);
while (!store.empty()) {
root = store.pop();
putLeftMost(list, store, root);
}
return list;
}
private void putLeftMost(List<Integer> list, Stack<TreeNode> store, TreeNode node) {
while (node != null) {
list.add(node.val);
if (node.right != null) {
store.push(node.right);
}
node = node.left;
}
}