Binary Tree Preorder Traversal 二叉树的前向遍历(迭代非递归版本)

二叉的前向遍历,即先输出当且节点的值,再遍历左子树,右子树。

同样的,先来造一个二叉树。

我先跟大家讲一个傻瓜式的做法(一开始我是这么做的,但发现速度上不行)

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;
        }
    }


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值