小米二轮面试编程题:二叉树的前序遍历,要求非递归方式

 

 

思路如下,使用栈,每次把当前节点入栈,然后右子节点入栈,左子节点入栈。

 

代码如下:

import java.util.*;
public class Solution {
    ArrayList<Integer> list = new ArrayList<Integer>();
    public ArrayList<Integer> preorderTraversal(TreeNode root) {
        if(root == null)
            return list;
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(root);
        while(!stack.isEmpty()){
            TreeNode temp = stack.pop();
            list.add(temp.val);
            if(temp.right != null)
                stack.push(temp.right);
            if(temp.left != null)
                stack.push(temp.left);
        }
        return list;
    }
}

 

posted on 2021-05-15 13:28  坚守梦想  阅读(110)  评论(0编辑  收藏  举报